mirrors
/
mmclassification
mirror of https://github.com/open-mmlab/mmclassification.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
mmclassification/mmcls/models/utils/augment/builder.py

9 lines
226 B
Python
Raw Normal View History

[Docs] Add Copyright information. (#413)
2021-08-17 19:52:42 +08:00
# Copyright (c) OpenMMLab. All rights reserved.
[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
from mmcv.utils import Registry, build_from_cfg
AUGMENT = Registry('augment')
def build_augment(cfg, default_args=None):
return build_from_cfg(cfg, AUGMENT, default_args)