pytorch-image-models/data/__init__.py

4 lines
239 B
Python

from data.dataset import Dataset
from data.transforms import transforms_imagenet_eval, transforms_imagenet_train, get_model_meanstd
from data.utils import create_loader
from data.random_erasing import RandomErasingTorch, RandomErasingNumpy