pytorch-image-models/timm/data
Ross Wightman 3bfd036b58 Add normalize flag to transforms factory, allow return of non-normalized native dtype torch.Tensors 2024-05-13 15:23:25 -07:00
..
_info Add teddy-bear class back to first 1000 classes of imagenet22k_ms_synsets (index 851) 2024-04-09 09:33:08 -07:00
readers Fiddling with iterator wrapping for HF ds streaming 2024-01-09 12:41:54 -08:00
__init__.py
auto_augment.py
config.py Add `in_chans` to data config helper. Fix #2021 2023-11-23 12:44:59 -08:00
constants.py
dataset.py Significant transforms, dataset, dataloading enhancements. 2024-01-08 09:38:42 -08:00
dataset_factory.py Significant transforms, dataset, dataloading enhancements. 2024-01-08 09:38:42 -08:00
dataset_info.py
distributed_sampler.py
imagenet_info.py
loader.py Pass train-crop-mode to create_loader/transforms from train.py args 2024-01-24 16:19:02 -08:00
mixup.py
random_erasing.py
real_labels.py
tf_preprocessing.py
transforms.py Small tweak of timm ToTensor for clarity 2024-02-10 14:57:40 -08:00
transforms_factory.py Add normalize flag to transforms factory, allow return of non-normalized native dtype torch.Tensors 2024-05-13 15:23:25 -07:00