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

Improve support for custom dataset label name/description through HF hub export, via pretrained_cfg

2023-02-08 08:29:20 -08:00

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_factory.py

Significant transforms, dataset, dataloading enhancements.

2024-01-08 09:38:42 -08:00

dataset_info.py

Improve support for custom dataset label name/description through HF hub export, via pretrained_cfg

2023-02-08 08:29:20 -08:00

dataset.py

Significant transforms, dataset, dataloading enhancements.

2024-01-08 09:38:42 -08:00

distributed_sampler.py

Significant model refactor and additions:

2022-02-28 13:56:23 -08:00

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

Add crop_mode for pretraind config / image transforms. Add support for dynamo compilation to benchmark/train/validate

2022-12-05 10:21:34 -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

transforms.py

Small tweak of timm ToTensor for clarity

2024-02-10 14:57:40 -08:00