otx.core.data#

Module for data related objects, such as OTXDataset, OTXDataModule, and Transforms.

Classes

OTXDataModule(task, config)

LightningDataModule extension for OTX pipeline.

OTXDatasetFactory()

Factory class for OTXDataset.

TransformLibFactory()

Factory class for transform.

class otx.core.data.OTXDataModule(task: OTXTaskType, config: DataModuleConfig)[source]#

Bases: LightningDataModule

LightningDataModule extension for OTX pipeline.

Constructor.

predict_dataloader() DataLoader[source]#

Get test dataloader.

setup(stage: str) None[source]#

Setup for each stage.

teardown(stage: str) None[source]#

Teardown for each stage.

test_dataloader() DataLoader[source]#

Get test dataloader.

train_dataloader() DataLoader[source]#

Get train dataloader.

val_dataloader() DataLoader[source]#

Get val dataloader.

property hparams_initial: AttributeDict#

The collection of hyperparameters saved with save_hyperparameters(). It is read-only.

The reason why we override is that we have some custom resolvers for DictConfig. Some resolved Python objects has not a primitive type, so that is not loggable without errors. Therefore, we need to unresolve it this time.

property tile_config: TileConfig#

Tiling configuration. It is a shortcut for self.config.tile_config.

class otx.core.data.OTXDatasetFactory[source]#

Bases: object

Factory class for OTXDataset.

classmethod create(task: OTXTaskType, dm_subset: DatasetSubset, mem_cache_handler: MemCacheHandlerBase, cfg_subset: SubsetConfig, cfg_data_module: DataModuleConfig) OTXDataset[source]#

Create OTXDataset.

class otx.core.data.TransformLibFactory[source]#

Bases: object

Factory class for transform.

classmethod generate(config: SubsetConfig) Transforms[source]#

Create transforms from factory.