You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
pytorch-image-models/data/__init__.py

4 lines
235 B

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