pytorch-image-models/timm/data/__init__.py

from .auto_augment import RandAugment, AutoAugment, rand_augment_ops, auto_augment_policy,\
    rand_augment_transform, auto_augment_transform
from .config import resolve_data_config, resolve_model_data_config
from .constants import *
from .dataset import ImageDataset, IterableImageDataset, AugMixDataset
from .dataset_factory import create_dataset
from .dataset_info import DatasetInfo, CustomDatasetInfo
from .imagenet_info import ImageNetInfo, infer_imagenet_subset
from .loader import create_loader
from .mixup import Mixup, FastCollateMixup
from .readers import create_reader
from .readers import get_img_extensions, is_img_extension, set_img_extensions, add_img_extensions, del_img_extensions
from .real_labels import RealLabelsImagenet
from .transforms import *
from .transforms_factory import create_transform
More dataset work including factories and a tensorflow datasets (TFDS) wrapper * 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 2021-01-15 17:26:20 -08:00			`from .auto_augment import RandAugment, AutoAugment, rand_augment_ops, auto_augment_policy,\`
			`rand_augment_transform, auto_augment_transform`
Cleanup resolve data config fns, add 'model' variant that takes model as first arg, make 'args' arg optional in original fn 2023-01-20 14:47:55 -08:00			`from .config import resolve_data_config, resolve_model_data_config`
More dataset work including factories and a tensorflow datasets (TFDS) wrapper * 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 2021-01-15 17:26:20 -08:00			`from .constants import *`
			`from .dataset import ImageDataset, IterableImageDataset, AugMixDataset`
			`from .dataset_factory import create_dataset`
Improve support for custom dataset label name/description through HF hub export, via pretrained_cfg 2023-02-08 07:56:19 -08:00			`from .dataset_info import DatasetInfo, CustomDatasetInfo`
Move ImageNet metadata (aka info) files to timm/data/_info. Add helper classes to make info available for labelling. Update inference.py for first use. 2023-02-06 17:41:07 -08:00			`from .imagenet_info import ImageNetInfo, infer_imagenet_subset`
Working on an implementation of AugMix with JensenShannonDivergence loss that's compatible with my AutoAugment and RandAugment impl 2019-12-19 18:16:18 -08:00			`from .loader import create_loader`
Final cleanup of mixup/cutmix. Element/batch modes working with both collate (prefetcher active) and without prefetcher. 2020-08-12 17:01:32 -07:00			`from .mixup import Mixup, FastCollateMixup`
Rename dataset/parsers -> dataset/readers, create_parser to create_reader, etc 2022-10-14 15:14:38 -07:00			`from .readers import create_reader`
			`from .readers import get_img_extensions, is_img_extension, set_img_extensions, add_img_extensions, del_img_extensions`
Working on feature extraction, interfaces refined, a number of models working, some in progress. 2020-06-29 17:45:38 -07:00			`from .real_labels import RealLabelsImagenet`
More dataset work including factories and a tensorflow datasets (TFDS) wrapper * 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 2021-01-15 17:26:20 -08:00			`from .transforms import *`
Improve image extension handling, add methods to modify / get defaults. Fix #1335 fix #1274. 2022-07-07 14:23:20 -07:00			`from .transforms_factory import create_transform`