mmpretrain

Commit Graph

Author	SHA1	Message	Date
Ma Zerun	f9eb9b409b	[Docs] Add Copyright information. (#413 )	2021-08-17 19:52:42 +08:00
Mingqiang Ning	4ebee155e8	fix a bug when samples_per_gpu==1 (#311 )	2021-06-30 20:57:21 +08:00
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
whcao	5e1a02103f	[Feature]Delete comments (#298 ) * add mytrain.py for test * test before layers * test attr in layers * test classifier * delete mytrain.py * delete comments	2021-06-12 21:45:22 +08:00
Miao Zheng	4ca21c7d03	[WIP] Refactoring weights initialization (#270 ) * [WIP] Refactoring weights initialization * fix lint and constant init cfg * fix pretrained bug * fix typo * fix isort * revise model utils	2021-06-10 10:54:34 +08:00
whcao	1cde6f6e65	[Feature] Add cutmix option (#198 ) * Add cutmix option * fix code style * add some annotations * add annotation about custom_hooks * check constraint of alpha > 0 * add test cutmix * fix code style * add cutmix to configs/models * add cutmix to configs/resnet * flake8 * empty	2021-04-14 21:27:42 +08:00
mzr1996	b7b520881f	Update CONTRIBUTING.md according to mmcv (#210 ) * Update CONTRIBUTING.md according to mmcv * Docstring formatting by docformatter * Update openmmlab website.	2021-04-14 21:22:37 +08:00
ftbabi	bdd6b01ae7	[Feature] Add "mixup" from Bag of Tricks (#160 ) * Add mixup option * Modify the structure of mixup and add configs * Clean configs * Add test for mixup and SoftCrossEntropyLoss * Add simple test for ImageClassifier * Fix bug in test_losses.py * Add assertion in CrossEntropyLoss	2021-02-25 14:06:58 +08:00
LXXXXR	a225cb6bdb	fix img_metas bug (#152 )	2021-01-26 11:24:08 +08:00
LXXXXR	07bb15e5fd	[Feature] Add heads and config for multilabel task (#145 ) * resolve conflicts add heads and config for multilabel tasks * minor change * remove evaluating mAP in head * add baseline config * add configs * reserve only one config * minor change * fix minor bug * minor change * minor change * add unittests and fix docstrings	2021-01-25 18:10:14 +08:00
yanglei	fbc68bf266	Add classifiers, heads, necks and losses	2020-07-07 19:32:06 +08:00

11 Commits (5cfaed6807c2cea2cd4f5f22e5fce8ab30bec7a5)