Ross Wightman cec70b6779

Merge pull request #2225 from huggingface/small_things ...

Small things

2024-07-25 20:29:13 -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

Fix pass through of input / target keys so ImageDataset readers so args work with hfds instead of just hfids (iterable)

2024-07-17 10:11:46 -07:00

__init__.py

…

auto_augment.py

Fix #1713 missed assignement in 3-aug level fn, fix few other minor lint complaints in auto_augment.py

2023-03-11 14:32:23 -08:00

config.py

…

constants.py

…

dataset_factory.py

…

dataset_info.py

…

dataset.py

Fix pass through of input / target keys so ImageDataset readers so args work with hfds instead of just hfids (iterable)

2024-07-17 10:11:46 -07:00

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

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:45:03 -08:00

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 'Maybe' PIL / image tensor conversions in case image alread in tensor format

2024-07-08 13:43:51 -07:00

transforms.py

Add 'Maybe' PIL / image tensor conversions in case image alread in tensor format

2024-07-08 13:43:51 -07:00