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/timm/data
Ross Wightman 7266c5c716
Merge branch 'main' into focalnet_and_swin_refactor
2 years ago
..
_info 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. 2 years ago
readers Allow using class_map functionality w/ IterableDataset (TFDS/WDS) to remap class labels 2 years ago
__init__.py Improve support for custom dataset label name/description through HF hub export, via pretrained_cfg 2 years ago
auto_augment.py Remove print from auto_augment 2 years ago
config.py Cleanup resolve data config fns, add 'model' variant that takes model as first arg, make 'args' arg optional in original fn 2 years ago
constants.py Add crop_mode for pretraind config / image transforms. Add support for dynamo compilation to benchmark/train/validate 2 years ago
dataset.py Allow using class_map functionality w/ IterableDataset (TFDS/WDS) to remap class labels 2 years ago
dataset_factory.py Allow using class_map functionality w/ IterableDataset (TFDS/WDS) to remap class labels 2 years ago
dataset_info.py Improve support for custom dataset label name/description through HF hub export, via pretrained_cfg 2 years ago
distributed_sampler.py
imagenet_info.py 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. 2 years ago
loader.py Add crop_mode for pretraind config / image transforms. Add support for dynamo compilation to benchmark/train/validate 2 years ago
mixup.py
random_erasing.py
real_labels.py 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. 2 years ago
tf_preprocessing.py Add crop_mode for pretraind config / image transforms. Add support for dynamo compilation to benchmark/train/validate 2 years ago
transforms.py Add crop_mode for pretraind config / image transforms. Add support for dynamo compilation to benchmark/train/validate 2 years ago
transforms_factory.py Add 3-Augment support to auto_augment.py, clean up weighted choice handling, and allow adjust per op prob via arg string 2 years ago