440 Commits

Author SHA1 Message Date
mzr1996
b35ee5d139 [Fix] Fix wrong default CLASSES settings in init_model. 2022-09-20 17:48:03 +08:00
Ma Zerun
e9e2d48cb2
[Improve] Update benchmark scripts (#1028)
* Update train benchmark scripts

* Add `--cfg-options` for dev scripts and enhance `--range`.

* Fix bug of regex expression.

* Fix minor bugs

* Update ShuffleNet configs

* Update rsb-a1 configs and label smooth loss mode.

* Update inference dev scripts

* From `mmengine` instead of `mmcv` import fileio.

* Fix lint

* Update pre-commit hook

* Use `use_sigmoid` option instead of "bce" mode in label smooth loss.
2022-09-20 15:50:21 +08:00
Hubert
e4e8047563
[Feature] Support Swin Transformer V2 for MMCLS 1.x. (#1029)
* [Feature] support swin v2 in mmcls 1.x

* fix configs

* fix clamp max

* Fix meshgrid

* Imporve swin v2

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-09-20 15:45:27 +08:00
Ezra-Yu
aaf127c5e1
[Feature] Add efficientformer Backbone for MMCls 1.x. (#1031)
* rebase

* update filename

* update URL

* update UT

* fix lint

* update head

* add efficientformer

* update filename

* update UT

* fix lint

* update configs

* rebase

* fix unit tests

* Fix comments and docs.

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-09-20 14:56:45 +08:00
Ezra-Yu
f1d2f50c21
[Feature] Add MobileOne Backbone For MMCls 1.x. (#1030)
* add mobileOne

* add train cfg

* update cfgs

* update URL

* update configs

* update inceptionv3 metafile

* add configs

* fix lint

* update checkpoint urls

* Update configs

* Update README

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-09-16 11:55:10 +08:00
Ma Zerun
61e9d890a6
[CI] Update circle-ci and github workflow. (#1018)
* Add deploy workflow.

* [CI] Update circle-ci and github workflow.

* Fix windows CI

* Update unit tests to save memory
2022-09-13 11:46:15 +08:00
Ma Zerun
85b1eae7f1
Bump to v1.0.0rc0 (#1007)
* Update docs.

* Update requirements.

* Update config readme and docstring.

* Update CONTRIBUTING.md

* Update README

* Update requirements/mminstall.txt

Co-authored-by: Yifei Yang <2744335995@qq.com>

* Update MMEngine docs link and add to readthedocs requirement.

Co-authored-by: Yifei Yang <2744335995@qq.com>
2022-08-31 23:57:51 +08:00
Ma Zerun
0b61b76325
[Docs] Add Migration documentation. (#997)
* [Docs] Add Migration documentation.

* Fix the train_cfg argument

* [WIP] Improve according to comments

* Imporve according to comments.
2022-08-31 13:31:48 +08:00
Ezra-Yu
e34588a315
[Docs] Refactor API docs. (#958)
* rebase

* update process

* add TODOs in moved classes

* update APIs

* modify the preprocess_cfg to data_preprocessor

* Fix api docs.

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-08-29 15:35:10 +08:00
Ezra-Yu
0e3e8e91cc
[Refactor] Refactor Config Doc. (#987)
* refactor config doc

* refine

* Fix the config docs.

* update CN config

* Fix some grammar erros.

* Fix Chinese docs.

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-08-29 11:10:05 +08:00
Ma Zerun
2b88df4484
[Refactor] Refactor the data flow. (#989)
* [Refactor] Refactor the data flow.

* Add comments about data preprocessor.

* Fix after mmengine folder structure refactoring.
2022-08-26 10:40:43 +08:00
Ma Zerun
b7d0d521eb
[Refactor] Use mmengine instead of mmcv and refactor some transforms. (#986)
* [Refactor] Refactor the rest data transforms and fix some docstring.

* Use `mmengine` instead of `mmcv`

* Refactor KFold dataset tools

* Fix docstring according to comments
2022-08-24 15:59:02 +08:00
Yixiao Fang
5665b8349a
[Refactor] Refactor import dependency. (#977)
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-08-22 15:17:33 +08:00
Ma Zerun
20f9ace5c5
[CI] Add Circle CI for mmcls 1.x. (#970)
* [CI] Add Circle CI for mmcls 1.x

* Fix circle-ci config

* Use `torch.cat` instead of `torch.hstack` to compat with PyTorch 1.6

* Compat the behavior of LongTensor in PyTorch 1.6

* Avoid random behavior in the unit test of Lighting

* Fix cuda ci

* Remove github workflow temporarily.
2022-08-22 15:02:08 +08:00
Yuan Liu
e4252d6848
[Feature]: Support BEiT Transformer layer. (#919)
* [Feature]: Add BEiT-style transformer encoder layer

* [Feature]: Add average token

* [Fix]: Fix lint

* [Fix]: Refactor CAE config

* [Fix]: Change cv2 backend to pillow backend

* [Fix]: Fix MAE and CAE reshape bug

* [Feature]: Add freeze vit layers

* [Feature]: Add mc

* [Fix]: Fix lint

* [Fix]: Fix dataset bug

* [Fix]: Delete cae selfsup config

* [Fix]: docstring

* [Refactor]: Add init_values to layer_scalue_init_value

* [Fix]: Refine the docstring of avg_token

* [Fix]: Call super init weight in beit attention

* [Fix]: remove mc

* [Fix]: Fix docstring

* [Fix]: Fix docstring

* [Fix]: Fix lint

* [Fix]: Fix init_value bug and change the logic of outputting cls token

* [Fix]: Fix docstring
2022-08-17 00:07:06 +08:00
Ma Zerun
b8b31e9343
[Improve] Multiple fixes. (#957)
* Rename some files in unit tests.

* Rename `pipelines` folder to `transforms`.

* Fix version requirement range.

* Remove `init_random_seed` and `set_random_seed`.

* Fix unit tests

* Rename `convert_models` to `model_converters`.
2022-08-09 15:58:08 +08:00
Ezra-Yu
624d22b441
[Refactor] Refactor Precise BN Hook. (#913) 2022-08-03 15:24:19 +08:00
Ma Zerun
43e60ad5a6
[Improve] Rename mmcls.data to mmcls.structures. (#941) 2022-07-29 14:18:13 +08:00
Ma Zerun
17b24a8230
[Improve] Update apis and packages. (#921)
* Remove mmcls/core folder

* update dependency version.

* Update `apis/inference.py` and demo.
2022-07-22 10:59:08 +08:00
Ezra-Yu
f37dc44a25 refactor UTs 2022-07-18 11:11:13 +08:00
Ezra-Yu
35cba87f82 refactor version and utils 2022-07-18 11:11:13 +08:00
Ezra-Yu
9118b97239 refactor evaluation 2022-07-18 11:11:13 +08:00
Ezra-Yu
6ac2f305a6 fix examples in doc 2022-07-18 11:11:13 +08:00
Ezra-Yu
317216efc6 fix visualizer registry bug 2022-07-18 11:11:13 +08:00
Ezra-Yu
1c82863449 fix lint 2022-07-18 11:11:13 +08:00
Ezra-Yu
6ed3860ea6 refactor UTs 2022-07-18 11:11:13 +08:00
Ezra-Yu
0f30c392a3 from core to engine 2022-07-18 11:11:13 +08:00
mzr1996
24bcf069f8 [Refactor] Refactor dev scripts 2022-07-18 11:11:13 +08:00
yingfhu
c992e24617 [Refactor] refactor ClassNumCheckHook 2022-07-18 11:11:13 +08:00
Ezra-Yu
e9342d9e4c Add multi_label heads 2022-07-18 11:11:13 +08:00
yingfhu
4fcd7ee072 Add EfficientNetRandomCrop and EfficientNetCenterCrop 2022-07-18 11:11:13 +08:00
mzr1996
12c982f939 [Refactor] Refactor RepeatAugSampler. 2022-07-18 11:11:13 +08:00
yingfhu
e8d69cf2ff [Fix] adjust default dataset path to empty string 2022-07-18 11:11:13 +08:00
Ezra-Yu
7bca2516f0 Add CUB dataset 2022-07-18 11:11:13 +08:00
Ezra-Yu
daa6167336 Add Lighting, ColorJitter and Albumentations 2022-07-18 11:11:13 +08:00
Ezra-Yu
a9057e88c4 Add multi label metrics 2022-07-18 11:11:13 +08:00
yingfhu
62b046521e [Refactor] refactor ViTHead, DeiTHead, ConformerHead, StackedHead 2022-07-18 11:11:13 +08:00
Ezra-Yu
a82de04b67 Enhance browse datset 2022-07-18 11:11:13 +08:00
mzr1996
daa716b112 Add some type hints 2022-07-18 11:11:13 +08:00
mzr1996
69e5ab065e Change "Default to" to "Defaults to" 2022-07-18 11:11:13 +08:00
mzr1996
7f13507fa8 Revert LongTensor check since cuda tensor is not long tensor 2022-07-18 11:11:13 +08:00
mzr1996
375fe68f12 Imporve according to comments 2022-07-18 11:11:13 +08:00
mzr1996
f0cab33e09 Force label in ClsDatasample long type. 2022-07-18 11:11:13 +08:00
mzr1996
f3299b4ca2 [Refactor] Refactor batch augmentations 2022-07-18 11:11:13 +08:00
mzr1996
dd660ed99e [Improve] Clean useless code and reduce unit tests memory usage. 2022-07-18 11:11:13 +08:00
mzr1996
cecff79a79 Rename 'feat' mode to 'tensor' mode 2022-07-18 11:11:13 +08:00
yingfhu
125b74d4ca [Refactor] Refactor MNIST and FashionMNIST dataset. 2022-07-18 11:11:13 +08:00
mzr1996
ee6e585e41 Mazerun/base classifier 2022-07-18 11:11:13 +08:00
Ezra-Yu
d6fa480915 [Refactor] Add VOC dataset and mutil_label dataset 2022-07-18 11:11:13 +08:00
mzr1996
522ab1fd84 [Feature] Add Visualization hook 2022-07-18 11:11:04 +08:00