modelzoo.vision.pytorch.input.classification.dataset_factory.VisionSubset#
- class modelzoo.vision.pytorch.input.classification.dataset_factory.VisionSubset[source]#
Bases:
torch.utils.data.Subset
Methods
transforms (callable, optional): A function/transforms that takes in
truncate_to_idx
- __call__(*args: Any, **kwargs: Any) Any #
Call self as a function.
- static __new__(cls, *args: Any, **kwargs: Any) Any #
- set_transforms(transforms=None, transform=None, target_transform=None)[source]#
- transforms (callable, optional): A function/transforms that takes in
an image and a label and returns the transformed versions of both.
- transform (callable, optional): A function/transform that takes in an PIL image
and returns a transformed version. E.g,
transforms.RandomCrop
- target_transform (callable, optional): A function/transform that takes in the
target and transforms it.