mirror of
https://github.com/huggingface/pytorch-image-models.git
synced 2025-06-02 22:31:44 +08:00
855d6cc217
* Add parser/dataset factory methods for more flexible dataset & parser creation * Add dataset parser that wraps TFDS image classification datasets * Tweak num_classes handling bug for 21k models * Add initial deit models so they can be benchmarked in next csv results runs
12 lines
553 B
Python
12 lines
553 B
Python
from .auto_augment import RandAugment, AutoAugment, rand_augment_ops, auto_augment_policy,\
|
|
rand_augment_transform, auto_augment_transform
|
|
from .config import resolve_data_config
|
|
from .constants import *
|
|
from .dataset import ImageDataset, IterableImageDataset, AugMixDataset
|
|
from .dataset_factory import create_dataset
|
|
from .loader import create_loader
|
|
from .mixup import Mixup, FastCollateMixup
|
|
from .parsers import create_parser
|
|
from .real_labels import RealLabelsImagenet
|
|
from .transforms import *
|
|
from .transforms_factory import create_transform |