mirrors
/
pytorch-image-models
mirror of https://github.com/huggingface/pytorch-image-models.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
pytorch-image-models/data/__init__.py

4 lines
239 B
Python
Raw Normal View History

Add some Nvidia performance enhancements (prefetch loader, fast collate), and refactor some of training and model fact/transforms
2019-03-11 05:23:16 +08:00
from data.dataset import Dataset
Improve creation of data pipeline with prefetch enabled vs disabled, fixup inception_res_v2 and dpn models
2019-03-12 13:17:42 +08:00
from data.transforms import transforms_imagenet_eval, transforms_imagenet_train, get_model_meanstd
from data.utils import create_loader
Add some Nvidia performance enhancements (prefetch loader, fast collate), and refactor some of training and model fact/transforms
2019-03-11 05:23:16 +08:00
from data.random_erasing import RandomErasingTorch, RandomErasingNumpy