mmpretrain/mmcls/models
whcao 3a08db9182
[Feature]Add augments to models/utils (#278)
* add mytrain.py for test

* test before layers

* test attr in layers

* test classifier

* delete mytrain.py

* add rand_bbox_minmax rand_bbox and cutmix_bbox_and_lam to BaseCutMixLayer

* add mixup_prob to BatchMixupLayer

* add cutmixup

* add cutmixup to __init__

* test classifier with cutmixup

* delete some comments

* set mixup_prob default to 1.0

* add cutmixup to classifier

* use cutmixup

* use cutmixup

* fix bugs

* test cutmixup

* move mixup and cutmix to augment

* inherit from BaseAugment

* add BaseAugment

* inherit from BaseAugment

* rename identity.py

* add @

* build augment

* register module

* rename to augment.py

* delete cutmixup.py

* do not inherit from BaseAugment

* add augments

* use augments in classifier

* prob default to 1.0

* add comments

* use augments

* use augments

* assert sum of augmentation probabilities should equal to 1

* augmentation probabilities equal to 1

* calculate Identity prob

* replace xxx with self.xxx

* add comments

* sync with augments

* for BC-breaking

* delete useless comments in mixup.py
2021-06-20 09:44:51 +08:00
..
backbones [Feature]Delete comments (#298) 2021-06-12 21:45:22 +08:00
classifiers [Feature]Add augments to models/utils (#278) 2021-06-20 09:44:51 +08:00
heads [Bug]Fix linearclshead (#307) 2021-06-16 00:37:16 +08:00
losses Refactor LabelSmoothLoss (#285) 2021-06-12 21:32:18 +08:00
necks GlabelAveragePooling support 1d, 2d and 3d by param, and add neck test (#236) 2021-05-10 15:00:50 +08:00
utils [Feature]Add augments to models/utils (#278) 2021-06-20 09:44:51 +08:00
__init__.py Add classifiers, heads, necks and losses 2020-07-07 19:32:06 +08:00
builder.py inherits mmcv registry (#252) 2021-05-14 23:36:56 +08:00