034919d032
[Feature] add eva02 backbone ( #1450 )
...
* [CI] Add test mim CI. (#879 )
* [CI] Add test mim CI. (#879 )
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* update
* update ci
* rebase
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* update
* update readme and configs
* update readme and configs
* refactore eva02
* [CI] Add test mim CI. (#879 )
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* update
* update ci
* rebase
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* update
* update readme and configs
* refactore eva02
* update readme and metafile
* update readme and metafile
* update readme and metafile
* update
* rename eva02
* rename eva02
* fix uts
* rename configs
---------
Co-authored-by: Ma Zerun <mzr1996@163.com>
Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
2023-05-06 19:28:31 +08:00
d9e561a09d
[Feature] Support dinov2 backbone ( #1522 )
...
* support dinov2 backbone
* update metafile and readme
* compatible to use_layer_scale
* update SwiGLUFFN
* add deprecation warning
* update
2023-05-05 16:59:37 +08:00
08dc8c75d3
[Refactor] Add selfsup algorithms. ( #1389 )
...
* remove basehead
* add moco series
* add byol simclr simsiam
* add ut
* update configs
* add simsiam hook
* add and refactor beit
* update ut
* add cae
* update extract_feat
* refactor cae
* add mae
* refactor data preprocessor
* update heads
* add maskfeat
* add milan
* add simmim
* add mixmim
* fix lint
* fix ut
* fix lint
* add eva
* add densecl
* add barlowtwins
* add swav
* fix lint
* update readtherdocs rst
* update docs
* update
* Decrease UT memory usage
* Fix docstring
* update DALLEEncoder
* Update model docs
* refactor dalle encoder
* update docstring
* fix ut
* fix config error
* add val_cfg and test_cfg
* refactor clip generator
* fix lint
* pass check
* fix ut
* add lars
* update type of BEiT in configs
* Use MMEngine style momentum in EMA.
* apply mmpretrain solarize
---------
Co-authored-by: mzr1996 <mzr1996@163.com>
2023-03-06 16:53:15 +08:00
c9670173aa
[Refactor] Move and refactor utils from mmselfsup. ( #1385 )
...
* add heads
* add losses
* fix
* remove mim head
* add modified backbones and target generators
* fix lint
* fix lint
* add heads
* add losses
* fix
* add data preprocessor from mmselfsup
* add ut for data prepocessor
* add GatherLayer
* add ema
* add batch shuffle
* add misc
* fix lint
* update
* update docstring
2023-02-28 17:04:40 +08:00
36bea13fca
[Refactor] Refactor ClsDatasample to a union DataSample. ( #1371 )
...
* [Refactor] Refactor ClsDatasample to a union DataSample.
* Add method
* Fix docstring
* Update docstring.
2023-02-23 10:07:53 +08:00
0979e78573
Rename the package name to `mmpretrain`.
2023-02-17 15:20:55 +08:00
97c4ae8805
[Improve] Update registries of mmcls. ( #1306 )
...
* [Improve] Update registries of mmcls.
* Update according to comments
2023-01-11 15:20:51 +08:00
060b0ed3b5
[Feature] Support ConvNeXt-V2 backbone. ( #1294 )
...
* [Feature] Support ConvNeXt-V2.
* Use registry of mmcls instead of mmengine.
* Add README.
* Add unit tests and docs.
2023-01-06 16:13:41 +08:00
29f066f7fb
[Improve] Speed up data preprocessor. ( #1064 )
...
* [Improve] Speed up data preprocessor.
* Add ClsDataSample serialization override functions.
* Add unit tests
* Modify configs to fit new mixup args.
* Fix `num_classes` of the ImageNet-21k config.
* Update docs.
2022-10-17 17:08:18 +08:00
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
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
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
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
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
43e60ad5a6
[Improve] Rename `mmcls.data` to `mmcls.structures`. ( #941 )
2022-07-29 14:18:13 +08:00
6ed3860ea6
refactor UTs
2022-07-18 11:11:13 +08:00
0f30c392a3
from core to engine
2022-07-18 11:11:13 +08:00
375fe68f12
Imporve according to comments
2022-07-18 11:11:13 +08:00
f3299b4ca2
[Refactor] Refactor batch augmentations
2022-07-18 11:11:13 +08:00
ee6e585e41
Mazerun/base classifier
2022-07-18 11:11:13 +08:00
c1534f9126
[Feature] Support resizemix. ( #676 )
...
* add resizemix
* skip torch.__version__ < 1.7.0
* Update mmcls/models/utils/augment/resizemix.py
Co-authored-by: Ma Zerun <mzr1996@163.com>
* Update mmcls/models/utils/augment/resizemix.py
Co-authored-by: Ma Zerun <mzr1996@163.com>
* resize -> F.interpolate
* fix docs
* fix test
* add Copyright
* add argument interpolation
Co-authored-by: Ma Zerun <mzr1996@163.com>
2022-03-07 12:11:20 +08:00
c708770b42
[Enhance] Support dynamic input shape for ViT-based algorithms. ( #706 )
...
* Move `resize_pos_embed` to `mmcls.models.utils`
* Refactor Vision Transformer
* Refactor DeiT
* Refactor MLP-Mixer
* Refactor Swin-Transformer
* Remove `indexing` arg
* Support dynamic inputs for t2t_vit
* Add copyright
* Fix bugs in swin transformer
* Add `pad_small_maps` option
* Update swin transformer
* Handle `attn_mask` in checkpoints of swin
* Imporve by comments
2022-03-03 13:10:12 +08:00
fcd57913ae
[Enhance] Upgrade isort pre-commit hooks. ( #687 )
...
* Use new version flake8 and isort hooks
* Fix missing copyright
2022-02-17 02:17:20 +08:00
94eb3ff83e
[Feature] Add Twins backbone and convert checkpoints. ( #642 )
...
* add twins backbone
* add position_encoding
* refactor twins
* Supplemental unit tests
* update docstring and readme
* update docstring and readme
* update docstring and readme
* update docstring
* update docstring
* update docstring
* update docstring
* remove note
* update doc and docstring
* update docstring
* update docstring
* use abstract pdf link and rename yamlfile
* Update model link
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-01-27 10:25:05 +08:00
d29037e8d1
[Enhance] Suport Mixup&Cutmix for multi-label task.
2022-01-21 11:30:58 +08:00
7dfc9e4a85
Bump version to v0.19.0 ( #623 )
2021-12-31 12:55:47 +08:00
0184527bd4
[Enhance] Enhence SE layer to support custom squeeze channels. ( #417 )
...
* add enhenced SE
* Update
* rm basechannel
* fix docstring
* Update se_layer.py
fix docstring
2021-08-20 13:31:44 +08:00
f9eb9b409b
[Docs] Add Copyright information. ( #413 )
2021-08-17 19:52:42 +08:00
1a7cebe4b9
[Refactor] Refactor unittest ( #321 )
...
* Refactor unit tests folder structure.
* Remove label smooth and Vit test in `test_classifiers.py`
* Rename test_utils in dataset to test_dataset_utils
* Split test_models/test_utils/test_utils.py to multiple sub files.
* Add unit tests of classifiers and heads
* Use patch context manager.
* Add unit test of `is_tracing`, and add warning in `is_tracing` if torch
verison is smaller than 1.6.0
2021-07-08 22:49:05 +08:00
zzc98
Yixiao Fang
Ma Zerun
Hubert
Ezra-Yu
Ezra-Yu
takuoko