51b7c754e7
[Fix] Fix the output position of Swin-Transformer. ( #947 )
...
* [Fix] Fix the output position of Swin-Transformer.
* Rename `downsample` argument to `do_downsample`.
2022-09-22 17:11:52 +08:00
034683cfcd
[Feature] Support MViT for MMCLS 1.x ( #1023 )
...
* [Feature] Support MViT for MMCLS 1.x
* Add unit tests
* Update documentation
2022-09-22 09:23:49 +08:00
f7191ca686
[Feature] Add ViT huge architecture. ( #1049 )
...
* [Feature]: Add ViT huge settings
* [Feature]: Add comments
2022-09-21 13:27:54 +08:00
81f10ccbb7
[Feature] Support EdgeNeXt for dev-1.x. ( #1037 )
...
* [Feature] add EdgeNeXt for dev-1.x
* [Feature] add update
* Imporve EdgeNeXt
* Fix div error in low pytorch version
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-09-21 13:27:04 +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
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
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
5665b8349a
[Refactor] Refactor import dependency. ( #977 )
...
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-08-22 15:17:33 +08:00
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
24bcf069f8
[Refactor] Refactor dev scripts
2022-07-18 11:11:13 +08:00
69e5ab065e
Change "Default to" to "Defaults to"
2022-07-18 11:11:13 +08:00
de002e455f
[Refactor] Use `MMLogger` from MMEngine in `get_logger` and `print_log`.
2022-07-18 10:53:27 +08:00
088d5b5add
[Refactor] Inherit all registries from MMEngine and use inherited registries.
2022-07-18 10:53:27 +08:00
3e198e63e8
[Docs] Update install tutorials. ( #854 )
...
* [Docs] Update install tutorials.
* [Docs] Improve dataset docs
* Add option to show the results in demo.
* fix typo
2022-06-01 18:31:57 +08:00
7c5ddb1e5b
Bump version to v0.23.0 ( #809 )
2022-05-01 21:58:33 +08:00
59292b315a
[Feature] Support DenseNet. ( #750 )
...
* init add densenet implementation
* Add config and converted models
* update meta
* add test for memory efficient
* Add docs
* add doc for jit
* Update checkpoint path
* Update readthedocs
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-04-28 13:06:03 +08:00
df6edd7f5a
[Feature] Support VAN. ( #739 )
...
* add van
* fix config
* add metafile
* add test
* model convert script
* fix review
* fix lint
* fix the configs and improve docs
* rm debug lines
* add VAN into api
Co-authored-by: Yu Zhaohui <1105212286@qq.com>
2022-04-28 09:35:17 +08:00
702c196514
[Docs] Refine API reference. ( #774 )
...
* [Docs] Refine API reference
* Add PoolFormer
2022-04-15 20:19:20 +08:00
29b882d4eb
Merge branch 'dev'
2022-04-15 20:10:45 +08:00
2eb27d0d4e
[Feature] Add PoolFormer backbone and checkpoints. ( #746 )
...
* add PoolFormer
* fix some typos in PoolFormer
* fix lint error
* modify out_indices and gap
* fix typo
* fix lint
* fix typo
* fix typo in poolforemr README
* fix lint
* Update some paths
* Refactor freeze_stages method
* Add unit tests
* Fix lint
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-04-15 11:35:15 +08:00
fbcf907dc6
[Feature] Support resize relative position embedding in `SwinTransformer`. ( #749 )
...
* [Feature]: Add resize rel pos embed
* [Refactor]: Create a separated resize_rel_pos_bias_table func
* [Refactor]: Refactor rel pos embed bias
* [Refactor]: Move interpolate into func
* Remove index buffer only when window_size changes
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-04-13 23:06:56 +08:00
922a31a68e
[Feature] ADD RepMLP Backbone and checkpoints. ( #709 )
...
* add repmlp
* refactor backbone code
* add checkpoint, readme, unit tests
* unchange demo
* improve docstring.
* update pic
* refactor patch embed
* refactor unit tests
* fix lint
* update tools
* update tools
* Update checkpoint path
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-03-30 20:40:25 +08:00
292127098f
[Feature] Add CSPNet and backbone and checkpoints ( #735 )
...
* add cspnet backbone
* try
* add backbones
* add cspnet
* add acc
* add unit tests
* chnage mode
* Update checkpoint path
* Fix typo
* Imporve CSPNet
* Update checkpoints
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-03-30 15:29:35 +08:00
1717efadbe
[Fix] Fix the discontiguous output feature map of ConvNeXt. ( #743 )
...
* [Fix] Fix the discontiguous output feature map of ConvNeXt.
* Add comment
2022-03-25 12:34:44 +08:00
04cb42a768
[Feature] Support ConvMixer. ( #716 )
...
* basic support for ConvMixer
* simplify
* add data pipeine config for timm
* Add model readme and metafile
* add unittest for convmixer
* add copyright
* modify
* add tests
* update model
* add conv2dAdaptivePadding replacement
* update model index
* fix comments
* Update checkpoint path
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-03-23 15:23:57 +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
5f7322c211
[Fix] Fix Conformer forward with irregular input size. ( #686 )
...
* Auto pad the feature map to be divisible by 4.
* Add unit tests
* Use `trans_down_stride` instead of 4.
2022-02-28 10:17:52 +08:00
7fcaedcbfb
[Feature] Add ResNetV1c. ( #692 )
...
* add ResNetV1c
* add unit tests
* fix lint
* update docstring
* fix lint
2022-02-23 11:36:33 +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
5de480ea9e
[Feature] Support HRNet and add pre-trained models ( #660 )
...
* Support HRNet
* Add HRNet configs
* Fix a bug in backward
* Add configs and update docs.
* Not use bias in conv before batch norm
* Defaults to use `norm_eval=False`
* Add unit tests and support out_channels in HRFuseScales
* Update checkpoint path
* Update docstring.
* Remove incorrect files
* Improve according to comments
2022-01-28 10:54:14 +08:00
dc456a0c2c
[Feature] Support ConvNeXt ( #670 )
...
* Support ConvNeXt
* Add configs of ConvNeXt
* Update dev scripts
* Update docs.
* Use new style README
* Add unit tests.
* Update README
* Imporve according to comments
* Modify refers to timm.
* Imporve according to comments
2022-01-28 10:36:45 +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
bd397f790f
[Enhance] Rewrite channel split operation in ShufflenetV2 ( #632 )
...
* replace chunk op
* shufflenetv2 config
2022-01-25 14:45:28 +08:00
d56170a734
[Feature] Support EfficientNet ( #649 )
...
* 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
* add efficient backbone
* add config
* add Edge Residual
* fix bug
* remove unnecessary files
* refactor
* add resize in crop to ensure crop size is output size
* fix bug and add comments
* test
* fix
* add more configs
* add more configs
* add more configs
* fix bug
* add model zoo
* fix
* reorganize code
* add edge tpu
* add edge tpu converter
* rename
* update readme
* reorganize code and config
* Rename configs of EfficientNet, and add metafile & model_zoo
* Remove `backend='pillow'`
* Add comments about EfficientNet-EdgeTPU
* Rename the convert tool of EfficientNet.
* Refactor EfficientNet and update docstring.
* Update EfficientNet-EdgeTPU config
* Fix unit tests
Co-authored-by: lixinran <lixr423@outlook.com>
Co-authored-by: lixinran <lixinran@sensetime.com>
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-01-25 12:14:17 +08:00
16864c7495
[Feature] Support features_only in TIMMBackbone ( #668 )
...
* Support features_only in TIMMBackbone
based on https://github.com/open-mmlab/mmsegmentation/pull/998
* update test for mmdet
* fix unit test for build_without_timm
* Update docstring
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-01-25 11:22:23 +08:00
0bbbb04429
[Enhancement] Improve the accuracy of ResNet ( #572 )
...
* add itp timm
* minor update
* minor update
* minor update
* add rep aug, minor update on configs
* minor update
* add target threshold
* add decaymulti
* minor update
* minor update
* add lbl smooth
* update lr
* reorganize config files and code
* minor bugfixes
* remove unused parts and minor fixes on cfg
* critical bugfix, add test and cfg update
* refactor code
* update doc string
* remove duplicate code
* refactor drop path in resnet
* rename
* Modify configs and add README&metafile
* Update metafile
Co-authored-by: mzr1996 <mzr1996@163.com>
2021-12-29 10:58:42 +08:00
131d8c6296
[Feature] Add CAM visualization tool ( #577 )
...
* add cam-grad tool
* refactor cam-grad tool
* add docs
* update docs
* Update docs and support Transformer
* remove pictures and use link
* replace example img and finish EN docs
* improve docs
* improve code
* Fix MobileNet V3 configs
* Refactor to support more powerful feature extraction.
* Add unit tests
* Fix unit test
* fix distortion of visualization exapmles in docs
* fix distortion
* fix distortion
* fix distortion
* merge master
* merge fix conficts
* Imporve the tool
* Support use both attribute name and index to get layer
* add default get_target-layers
* add default get_target-layers
* update docs
* update docs
* add additional printt info when not using target-layers
* Imporve docs
* Fix enumerate list.
Co-authored-by: mzr1996 <mzr1996@163.com>
2021-12-23 18:53:40 +08:00
f9a2b04cee
[Feature] Add DeiT backbone and checkpoints. ( #576 )
...
* Support DeiT backbone.
* Use hook to automatically resize pos embed
* Update ViT training setting
* Add deit configs and update docs
* Fix vit arch assertion
* Remove useless init function
* Add unit tests.
* Fix resize_pos_embed for DeiT
* Improve according to comments.
2021-12-15 22:44:57 +08:00
18f6bb0b10
[Feature] Implement the conformer backbone. ( #494 )
...
* implement the conformer
* format code style
* format code style
* reuse the TransformerEncoderLayer in the vision_transformer.py
* Modify variable name
* delete unused params
* Remove warning info in Conformer head since it already exists in
Conformer.
* Rename some variables
* Add unit tests
* Use `getattr` instead of `get_submodule`.
* Remove some useless layers
* Refactor conformer and add configs
* Update configs and add metafile.
* Fix unit tests
* Update README
Co-authored-by: mzr1996 <mzr1996@163.com>
2021-12-07 14:00:17 +08:00
0aa789f3c3
[Feature] Add the frozen function for Swin Transformer model. ( #574 )
...
* Add the frozen function for Swin Transformer model
* add frozen parameter for swin transformer model
* add norm_eval parameter
* Delete =11.1
* Delete =418,driver
* delete _BatchNorm
* remove LayerNorm , add _BatchNorm
* unifying the style of frozen function refer ResNet
* Improve docs and add unit tests.
Co-authored-by: cxiang26 <cq.xiang@foxmail.com>
Co-authored-by: mzr1996 <mzr1996@163.com>
2021-12-07 11:58:14 +08:00
f2c1c57398
[Docs] Fix docs build dependency. ( #584 )
...
* Fix docs.
* Add MMHuman3D
2021-12-07 11:16:33 +08:00
c090d3f7fd
[Fix] Fix a bug for multiple output in swin transformer. ( #571 )
...
* Update swin_transformer.py
fix bug;
layer_norm 0,1,2,3 will be the same and resulted in an error when out_indices dict exceed one element.
* fix bug, layer_norm 0,1,2,3
* Add unit tests for multiple out_indices
Co-authored-by: cxiang26 <cq.xiang@foxmail.com>
Co-authored-by: mzr1996 <mzr1996@163.com>
2021-12-03 16:11:41 +08:00
9d9dce69ad
[Feature] Support using checkpoint in Swin Transformer to save memory. ( #557 )
...
* add checkpoint in swin backbone
* add checkpoint in swin backbone
2021-12-01 11:49:00 +08:00
fc8adbc149
[Enhance] Add metafile, readme and converted models for MLP-Mixer ( #539 )
...
* add pth converter
* minor update on config files, add metafile and readme
* add missing readme and minor fixes
* minor fixes
* Update config names and checkpoint download link
* Update model_zoo.md
Co-authored-by: mzr1996 <mzr1996@163.com>
2021-11-24 19:04:19 +08:00
9f31fc5390
[Refactor] Use `reset_classifier` to remove head of timm backbones. ( #534 )
...
* refactor timm backbone
* fix comment
2021-11-16 16:24:55 +08:00
3eafc5cbe1
[Feature] Add MLP Mixer Backbone. ( #528 )
...
* Add Mlp Mixer
* Fix token_mix FNN add_identity and add out_indices
* Fix mlp_mixer unit test
2021-11-15 18:25:22 +08:00
fffa30dd48
[Feature] Add Tokens-to-Token ViT backbone and converted checkpoints. ( #467 )
...
* add t2t backbone
* register t2t_vit
* add t2t_vit config
* [Temp] Align posterize transform with timm.
* Fix lint
* Refactor t2t-vit
* Add config for t2t-vit
* Add metafile and README for t2t-vit
* Add unit tests
* configs
* Update metafile and README
* Improve docstring
* Fix batch size which should be 8x64 instead of 8x128
* Fix typo
* Update model zoo
* Update training augments config.
* Move some arguments of T2TModule to T2TViT
* Update docs.
* Update unit test
Co-authored-by: HIT-cwh <2892770585@qq.com>
2021-10-29 10:37:16 +08:00
671414becb
[Fix] Fix missing import `Compose` and `Normalize`.
...
* Fixed missing import 'Compose'
* Fixed mistype `Compose` in `mmcls/datasets/__init__.py`
* Fixed missing import `Normalize`
* [Docs] Fix typos in doctest
* [Fix] Sort import module
2021-10-28 15:21:05 +08:00
Ma Zerun
Yuan Liu
Songyang Zhang
Hubert
Ezra-Yu
Yixiao Fang
yingfhu
Ezra-Yu
takuoko
Weihao Yu
Zhicheng Chen
Yosuke Shinya
Zhiliang Peng
fangxu
xcnick
imyhxy