Commit Graph

635 Commits (b7d0d521eb12bfa3666dd7a633d534c6ca0de857)
 

Author SHA1 Message Date
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
Ma Zerun 7f0502867b
[Improve] Add `--amp`, `--resume` and `--auto-scale-lr` options in `tools/train.py`. (#914) 2022-07-21 17:34:04 +08:00
mzr1996 0c7a04b1c7 Fix lint 2022-07-18 14:14:31 +08:00
mzr1996 735a3ee11f Update auto_scale_lr fields 2022-07-18 11:11:13 +08:00
yingfhu a667b488ae minor fix 2022-07-18 11:11:13 +08:00
yingfhu ce81a07059 [Refactor] add auto_scale_lr 2022-07-18 11:11:13 +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
mzr1996 c0f3ba68a8 Add MAE version ViT-base training results and fix some errors in configs. 2022-07-18 11:11:13 +08:00
Ezra-Yu d8f556668e Refactor vis scheduler tool 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
mzr1996 5a3e736b4e [Feature] Implement offline evaluation. 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
yingfhu c78b5597d8 [Refactor] refactor randomCrop, randomResizeCrop and CenterCrop。 2022-07-18 10:57:17 +08:00
mzr1996 6563f5f448 Update autoaug and randaug configs 2022-07-18 10:53:56 +08:00
mzr1996 05e4c13343 Improve according to comments 2022-07-18 10:53:56 +08:00