==================
DIST -> DataLoader
==================

General DataLoader
==================

General class for creating Datasets. It works by detecting all specified
file formats in the given directory and return the index list.

.. autoclass:: tardis_em.dist_pytorch.datasets.dataloader.BasicDataset

Specialized DataLoader's
========================

Filament structure DataSet
--------------------------

.. autoclass:: tardis_em.dist_pytorch.datasets.dataloader.FilamentDataset

PartNet synthetic DataSet

.. autoclass:: tardis_em.dist_pytorch.datasets.dataloader.PartnetDataset

ScanNet V2 synthetic DataSet
----------------------------

.. autoclass:: tardis_em.dist_pytorch.datasets.dataloader.ScannetDataset

ScanNet V2 with RGB values synthetic DataSet
--------------------------------------------

.. autoclass:: tardis_em.dist_pytorch.datasets.dataloader.ScannetColorDataset

Stanford S3DIS DataSet
----------------------

.. autoclass:: tardis_em.dist_pytorch.datasets.dataloader.Stanford3DDataset

Helper Functions
================

Build dataloader
-----------------

.. autofunction:: tardis_em.dist_pytorch.datasets.dataloader.build_dataset

Point Cloud Augmentation
------------------------

.. automodule:: tardis_em.dist_pytorch.datasets.augmentation

Patch dataset
-------------

.. automodule:: tardis_em.dist_pytorch.datasets.patches