pytorch-image-models/data/__init__.py

4 lines
235 B
Python

from data.dataset import Dataset
from data.transforms import transforms_imagenet_eval, transforms_imagenet_train
from data.utils import fast_collate, PrefetchLoader
from data.random_erasing import RandomErasingTorch, RandomErasingNumpy