datumaro.plugins.framework_converter#

Classes

`DmTfDataset`(dataset, subset, task[, ...])
`DmTorchDataset`(dataset, subset, task[, ...])
`FrameworkConverter`(dataset, subset, task)
`FrameworkConverterFactory`()

class datumaro.plugins.framework_converter.FrameworkConverterFactory[source]#

Bases: object

static create_converter(framework)[source]#

class datumaro.plugins.framework_converter.FrameworkConverter(dataset, subset, task)[source]#

Bases: object

to_framework(framework, **kwargs)[source]#

class datumaro.plugins.framework_converter.DmTorchDataset(dataset: Dataset, subset: str, task: str, transform: Optional[Callable] = None, target_transform: Optional[Callable] = None)[source]#: Bases: _MultiFrameworkDataset, Dataset

class datumaro.plugins.framework_converter.DmTfDataset(dataset: Dataset, subset: str, task: str, output_signature: Optional[tuple] = None)[source]#

Bases: _MultiFrameworkDataset

create() → DatasetV2[source]#

repeat(count=None) → DatasetV2[source]#

batch(batch_size, drop_remainder=False) → DatasetV2[source]#