TorchDatasetMapperΒΆ

Module providing implementation of a PyTorch dataset mapper for Ikomia dataset structure. It converts the Ikomia dataset structure to a PyTorch compatible dataset so that it is possible to use Ikomia dataset loaders with PyTorch Deep Learning models.

Import

from ikomia.dnn.torch import datasetmapper

Classes

TorchDatasetMapper(ik_dataset, ...)

Class derived from torch.utils.data.Dataset.