Commit Graph

408 Commits (b486bb23f8870c10df14b2f5987c15bf9d3f979b)
 

Author SHA1 Message Date
Ma Zerun 19cfb25e5e
Use parameter default value to control default behavior of init_cfg in (#319)
`LinearClsHead` and `MultiLabelLinearClsHead`

And remove the verbose `_init_layers` method of `LinearClsHead` and
`MultiLabelLinearClsHead`.
2021-06-30 19:13:27 +08:00
Ma Zerun db5502f525
Fix config link of SeResNet README. (#326) 2021-06-29 23:19:58 +08:00
Ma Zerun 65410b05ad
Fix Mobilenetv3 structure and add pretrained model (#291)
* Refactor Mobilenetv3 structure and add ConvClsHead.

* Change model's name from 'MobileNetv3' to 'MobileNetV3'

* Modify configs for MobileNetV3 on CIFAR10.

And add MobileNetV3 configs for imagenet

* Fix activate setting bugs in MobileNetV3.

And remove bias in SELayer.

* Modify unittest

* Remove useless config and file.

* Fix mobilenetv3-large arch setting

* Add dropout option in ConvClsHead

* Fix MobilenetV3 structure according to torchvision version.

1. Remove with_expand_conv option in InvertedResidual, it should be decided by channels.

2. Revert activation function, should before SE layer.

* Format code.

* Rename MobilenetV3 arch "big" to "large".

* Add mobilenetv3_small torchvision training recipe

* Modify default `out_indices` of MobilenetV3, now it will change
according to `arch` if not specified.

* Add MobilenetV3 large config.

* Add mobilenetv3 README

* Modify InvertedResidual unit test.

* Refactor ConvClsHead to StackedLinearClsHead, and add unit tests.

* Add unit test for `simple_test` of `StackedLinearClsHead`.

* Fix typo

Co-authored-by: Yidi Shao <ydshao@smail.nju.edu.cn>
2021-06-27 23:19:36 +08:00
Ma Zerun 53c0df271f
Fix magnitude_std bug in RandAugment, and update unit tests. (#309) 2021-06-21 11:25:11 +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 b9879a4667
[Bug]Fix linearclshead (#307)
* add mytrain.py for test

* test before layers

* test attr in layers

* test classifier

* delete mytrain.py

* fix init_cfg bug
2021-06-16 00:37:16 +08:00
whcao d62f198d2b
[Feature]Support custom hooks (#305)
* add mytrain.py for test

* test before layers

* test attr in layers

* test classifier

* delete mytrain.py

* register custom_hooks in runner

* set custom_hooks_config to cfg.get(custom_hooks, None)
2021-06-15 21:09:58 +08:00
whcao 438c9da6eb
[Feature]Fix linear cls head (#303)
* add mytrain.py for test

* test before layers

* test attr in layers

* test classifier

* delete mytrain.py

* move init_cfg to parameter

* isort

* Use a sentinel value to denote the default init_cfg
2021-06-15 21:08:30 +08:00
AllentDan a24a9f6faa
[Fix] Build compatible with low pytorch versions (#301)
* add version compatible for torchscript

* doc

* doc again

* fix lint

* fix lint isort
2021-06-14 23:25:35 +08:00
WRH b99bd4fa88
Fix bug for CPU training (#286)
* remove MMDataParallel when using cpu

* support cpu testing

* fix lint
2021-06-12 22:26:33 +08:00
AllentDan c2f01e0dcd
[Feature] Add torchscript deployment (#279)
* add torchscript deploy

* fix lint

* add check and delete \
2021-06-12 21:50:48 +08:00
q.yao dbddde52ef
[Feature] TensorRT test tools. (#284)
* first commit

* update resnext result

* update docs

* update docstring
2021-06-12 21:47:10 +08:00
David de la Iglesia Castro e2507cae97
Add mmcls2torchserve (#292)
* Add mmcls2torchserve

* Update docs
2021-06-12 21:45:45 +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
Ma Zerun 84a939f858
Refactor LabelSmoothLoss (#285)
* Refector label smooth loss, now support mode `original`, `classy_vision`
and `multi_label`.

* Add unittests for label smooth loss.

* Improve docstring of LSR
2021-06-12 21:32:18 +08:00
LXXXXR 6de635a81c
[Bug] Missing test data when num_imgs can not be evenly divided by num_gpus (#299)
* fix bug in test

* remove unneccesary code
2021-06-11 15:53:23 +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
LXXXXR 5066e32306
dump config before training (#282) 2021-06-04 16:26:13 +08:00
Ma Zerun 9f7168e770
Add RegNetX configs (#269)
* Add RegNetX configs

* Fix typo in RegNet readme.

* Use comment to mark tag.
2021-06-04 14:29:13 +08:00
Ma Zerun b95caca0d5
Redirect model_zoo.md and format Chinese documents. (#290)
* Redirect model_zoo.md in Chinese Readme to English ver.

* Format Chinese documents.
2021-06-03 16:49:53 +08:00
LXXXXR c68bf9f2ba
fix resnetv1d configs and update ckpts (#288) 2021-06-03 16:30:29 +08:00
LXXXXR 27a49a9646
bump version to v0.12.0 (#287) 2021-06-03 11:42:34 +08:00
LXXXXR df3d2d47f0
update resnext ckpts (#283) 2021-06-01 14:57:46 +08:00
LXXXXR 5ff158cec9
[Bug] Fix Cifar color channel (#280)
* fix bug

* updatedoc accordingly
2021-05-31 14:15:22 +08:00
LXXXXR 2c9e12f850
[Feature] Add an argument `efficientnet_style` to `RandomResizedCrop` and `CenterCrop` (#268)
* add config for resnest test

* fix config

* add label smoothing

* add memcached

* minor fix

* fix bug

* fix config

* add config

* minor fix

* fix configs

* use EResize

* change interpolation

* add more configs

* add docsting

* add unittest

* remove unnecessary changes

* minor fix

* add more docstring

* fix linting

* refactor

* add resize in crop to ensure crop size is output size

* fix bug and add comments

* fix bug
2021-05-31 14:10:57 +08:00
Mingqiang Ning 9a66113d20
add attribute 'test' in dict 'data' (#264)
* add attribute 'test' in dict 'data'

* Update lenet5_mnist.py

 delete spaces around '='.

* init

* Delete custom directory
2021-05-31 10:19:27 +08:00
Ma Zerun 4524327a4a
[Docs] Fix typo, improve and add Chinese translation of `data_pipeline.md` and `new_modules.md` (#265)
* Improve docs, including `new_modules.md` and `data_pipeline.md`

* Add Chinese docs `new_modules.md` and `data_pipeline.md`
2021-05-29 10:48:31 +08:00
LXXXXR b23b319f56
fix version (#276) 2021-05-29 10:47:16 +08:00
LXXXXR bd9411d743
[Bug] Download dataset only on rank 0 (#273)
* only download dataset on rank 0

* download only on rank 0

* fix bug

* fix error message
2021-05-29 10:45:58 +08:00
Ma Zerun a06827bc08
Build chinese translation in readthedocs (#267)
* change document layout

* Add Chinese docs build file

* Add switch language option
2021-05-29 10:00:37 +08:00
LXXXXR 98b6c4afa4
update cifar config (#275) 2021-05-27 15:20:25 +08:00
Yinhao Li b67a11c548
Fix kwargss to kwargs. (#274) 2021-05-26 19:27:30 +08:00
LXXXXR dac090162d
Bump version to v0.11.1 (#256)
* bump version to v0.11.1

* minor fix

* minor fix

* minor fix

* minor fix
2021-05-21 16:36:08 +08:00
Y. Xiong 82e3937174
[Fix] Only allow directory operation when rank==0 when testing (#258)
* only allow dir operation when rank==0

* move check dir to multi_gpu_test
2021-05-21 14:04:46 +08:00
Ma Zerun 09597e5a4c
Add transform `RandomErasing` (#248)
* Add transform `RandomErasing`.

* Add unittests of `RandomErasing`

* Fix typo in docstring

* Improve docstring and unittests.
2021-05-19 22:35:26 +08:00
LXXXXR ace09510f7
[Bug]Fix typo (#255)
* fix typo

* rerun ci
2021-05-17 10:42:31 +08:00
LXXXXR dc296f64c6
[Fix] Fix multi-node test tmp dir (#251)
* fix multi-node test tmp dir

* fix mmcv version
2021-05-16 21:53:13 +08:00
Wenwei Zhang 5ee08767f2
inherits mmcv registry (#252) 2021-05-14 23:36:56 +08:00
LXXXXR 8c90a879ce
[Fix] Fix magnitude_range in RandAug (#249)
* add increasing in solarize and posterize

* fix linting

* Revert "add increasing in solarize and posterize"

This reverts commit 128af36e9b.

* revise according to comments
2021-05-12 15:21:55 +08:00
LXXXXR f415c49bc3
fix MANIFEST.in (#250) 2021-05-12 15:04:58 +08:00
mzr1996 9e93feabac
[Docs] Fix error in new_dataset.md and add Chinese translation of finture.md, new_dataset.md (#243)
* Fix error in new_dataset.md

* Add Chinese Translation of finture.md, new_dataset.md
2021-05-10 17:17:37 +08:00
mzr1996 a3b8d6015d
[Feature] Add RandAUG magnitude noise (#240)
* Add paramater magnitude_std in RandAugment to allow randomly movement of magnitude_value

* Add unittest for magnitude_std

* Improve docstring of magnitude_std
2021-05-10 17:13:41 +08:00
Y. Xiong 22b40696d6
add tempdir check (#242) 2021-05-10 17:12:43 +08:00
Forsaken 0a92a81198
Update analyze_results.py (#237)
correct spelling mistake(sucess->success)
2021-05-10 15:04:08 +08:00
RangiLyu 6a50a2864b
fix typo in unit test (#238) 2021-05-10 15:03:39 +08:00
mzr1996 8128900a12
GlabelAveragePooling support 1d, 2d and 3d by param, and add neck test (#236)
* GlabelAveragePooling support 1d, 2d and 3d by param, and add neck test

* Imporve neck test

* Change 'mode' attribute in GAP to 'dim', and add docstring
2021-05-10 15:00:50 +08:00
whcao b30f79ea4c
[Feature]Modify Parameters Passing in models.heads (#239)
* add mytrain.py for test

* test before layers

* test attr in layers

* test classifier

* delete mytrain.py

* set cal_acc in ClsHead defaults to False

* set cal_acc defaults to False

* use *args, **kwargs instead

* change bs16 to 3 in test_image_classifier_vit

* fix some comments

* change cal_acc=True

* test LinearClsHead
2021-05-10 14:56:55 +08:00
LXXXXR 37167158e7
bump version to v0.11.0 (#233) 2021-05-01 22:26:39 +08:00
LXXXXR 761c95925d
support mmcli (#232) 2021-05-01 22:24:16 +08:00
QingChuanWS f970a36898
update pytorch2onnx.md table (#229) 2021-04-30 17:18:06 +08:00