Ross Wightman 670c61b28f Some cutmix/mixup cleanup/fixes

2020-02-17 11:00:54 -08:00

..

__init__.py

Working on CutMix impl as per #8, integrating with Mixup, currently experimenting...

2020-02-16 20:09:11 -08:00

auto_augment.py

Update AugMix, JSD, etc comments and references

2020-01-11 12:02:05 -08:00

config.py

A number of tweaks to arguments, epoch handling, config

2019-06-28 13:49:20 -07:00

constants.py

Big re-org, working towards making pip/module as 'timm'

2019-06-19 17:20:51 -07:00

dataset.py

Working on CutMix impl as per #8, integrating with Mixup, currently experimenting...

2020-02-16 20:09:11 -08:00

distributed_sampler.py

Big re-org, working towards making pip/module as 'timm'

2019-06-19 17:20:51 -07:00

loader.py

Merge pull request #74 from rwightman/augmix-jsd

2020-01-11 12:04:29 -08:00

mixup.py

Some cutmix/mixup cleanup/fixes

2020-02-17 11:00:54 -08:00

random_erasing.py

Add SplitBatchNorm. AugMix, Rand/AutoAugment, Split (Aux) BatchNorm, Jensen-Shannon Divergence, RandomErasing all working together

2020-01-05 19:58:59 -08:00

tf_preprocessing.py

Add MixNet (https://arxiv.org/abs/1907.09595) with pretrained weights converted from Tensorflow impl

2019-07-25 11:42:01 -07:00

transforms_factory.py

Indentation mistake. Fixes #81

2020-02-04 22:56:00 -08:00

transforms.py

Working on an implementation of AugMix with JensenShannonDivergence loss that's compatible with my AutoAugment and RandAugment impl

2019-12-20 23:04:11 -08:00