bedf4e9f64
[Refactor] Update analysis tools and documentations. ( #1359 )
...
* [Refactor] Update analysis tools and documentations.
* Update migration.md and add unit test.
* Fix print_config.py
2023-02-15 10:28:08 +08:00
b4ee9d2848
[Feature] Support calculate confusion matrix and plot it. ( #1287 )
...
* [Feature] Support calculate confusion matrix and plot it.
* Fix keepdim
* Update confusion_matrix tools and the plot graph.
* Revert accidental modification.
* Update docstring
* Move confusion matrix tool to
2023-02-14 12:58:11 +08:00
7ec6062415
[Refactor] Unify the `--out` and `--dump` in `tools/test.py`. ( #1307 )
2023-02-09 14:05:03 +08:00
6b9e2b55dd
[Feature] Support LeViT backbone. ( #1238 )
...
* 网络搭建完成、能正常推理
* 网络搭建完成、能正常推理
* 网络搭建完成、能正常推理
* 添加了模型转换未验证，配置文件 但有无法运行
* 模型转换、结构验证完成，可以推理出正确答案
* 推理精度与原论文一致 已完成转化
* 三个方法改为class 暂存
* 完成推理精度对齐 误差0.04
* 暂时使用的levit2mmcls
* 训练跑通，训练相关参数未对齐
* '训练相关参数对齐'参数'
* '修复训练时验证导致模型结构改变无法复原问题'
* '修复训练时验证导致模型结构改变无法复原问题'
* '添加mixup和labelsmooth'
* '配置文件补齐'
* 添加模型转换
* 添加meta文件
* 添加meta文件
* 删除demo.py测试文件
* 添加模型README文件
* docs文件回滚
* model-index删除末行空格
* 更新模型metafile
* 更新metafile
* 更新metafile
* 更新README和metafile
* 更新模型README
* 更新模型metafile
* Delete the model class and get_LeViT_model methods in the mmcls.models.backone.levit file
* Change the class name to Google Code Style
* use arch to provide default architectures
* use nn.Conv2d
* mmcv.cnn.fuse_conv_bn
* modify some details
* remove down_ops from the architectures.
* remove init_weight function
* Modify ambiguous variable names
* Change the drop_path in config to drop_path_rate
* Add unit test
* remove train function
* add unit test
* modify nn.norm1d to build_norm_layer
* update metafile and readme
* Update configs and LeViT implementations.
* Update README.
* Add docstring and update unit tests.
* Revert irrelative modification.
* Fix unit tests
* minor fix
Co-authored-by: mzr1996 <mzr1996@163.com>
2023-01-17 17:43:42 +08:00
c98dc4555c
[Feature] Support VIG Backbone. ( #1304 )
...
* 添加vig源文件
* 某些模块修改到mmcls风格
* 修改到mmcls风格
* 修改
* 添加VIG模型及源文件
* update model file
* update model file and config
* change class name and some variable name
* change class name and some variable name
* update
* update
* change nn.BatchNorm to mmcv.cnn.build_norm_layer
* update
* change nn.Seq to mmcls
* change backbone to stage_blocks
* add vig_head
* update
* update config file
* update
* add readme and metafile
* update model-index.yml
* update model file
* rename config file and add docstring
* variable rename
* update readme and metafile
* update readme
* update
* Update VIG backbone implementation and docs.
* Fix configs.
Co-authored-by: mzr1996 <mzr1996@163.com>
2023-01-17 16:55:56 +08:00
c73a5a8b15
[Fix] Fix bug in test tools. ( #1309 )
...
* fix append
* Update test.py
* Update test.py
2023-01-12 14:18:00 +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
74743ef588
[Feature] [CodeCamp #68 ] Add EfficientnetV2 Backbone. ( #1253 )
...
* add efficientnet_v2.py
* add efficientnet_v2 in __init__.py
* add efficientnet_v2_s base config file
* add efficientnet_v2 config file
* add efficientnet_v2 config file
* update tuple output
* update config file
* update model file
* update model file
* update model file
* update config file
* update model file
* update config file
* update model file
* update model file
* update model file
* update model file
* update model file
* update config file
* update config file
* update model file
* update model file
* update model file
* update model file
* update model config file
* Update efficientnet_v2.py
* add config file and modify arch
* add config file and modify arch
* add the file about convert_pth from timm to mmcls
* update efficientnetv2 model file with mmcls style
* add the file about convert_pth from timm to mmcls
* add the file about convert_pth from timm to mmcls
* update convert file
* update model file
* update convert file
* update model file
* update model file
* update model file
* add metefile and README
* Update tools/model_converters/efficientnetv2-timm_to_mmcls.py
Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
* update model file and convert file
* update model file and convert file
* update model file and convert file
* update model file and convert file
* update model file
* update model file
* update model file
* update config file and README file
* update metafile
* Update efficientnetv2_to_mmcls.py
* update model-index.yml
* update metafile.yml
* update b0 and s train pipeline
* update b0 and s train pipeline
* update b0 and s train pipeline
* add test_efficientnet_v2
* update test_efficientnet_v2
* update model file docs
* update test_efficientnet_v2
* update test_efficientnet_v2
* add efficientnet_v2.py
* add efficientnet_v2 in __init__.py
* add efficientnet_v2_s base config file
* add efficientnet_v2 config file
* add efficientnet_v2 config file
* update tuple output
* update config file
* update model file
* update model file
* update model file
* update model file
* update config file
* update config file
* update model file
* update model file
* update model file
* update model file
* update model file
* update config file
* update config file
* update model file
* update model file
* update model file
* update model file
* update model config file
* Update efficientnet_v2.py
* add config file and modify arch
* add config file and modify arch
* add the file about convert_pth from timm to mmcls
* update efficientnetv2 model file with mmcls style
* add the file about convert_pth from timm to mmcls
* add the file about convert_pth from timm to mmcls
* update convert file
* update model file
* update convert file
* update model file
* update model file
* update model file
* add metefile and README
* Update tools/model_converters/efficientnetv2-timm_to_mmcls.py
Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
* update model file and convert file
* update model file and convert file
* update model file and convert file
* update model file and convert file
* update model file
* update model file
* update model file
* update config file and README file
* update metafile
* Update efficientnetv2_to_mmcls.py
* update model-index.yml
* update metafile.yml
* update b0 and s train pipeline
* update b0 and s train pipeline
* update b0 and s train pipeline
* add test_efficientnet_v2
* update test_efficientnet_v2
* update model file docs
* update test_efficientnet_v2
* update test_efficientnet_v2
* pass pre-commit hook
* refactor efficientnetv2
* refactor efficientnetv2
* update readme, metafile and weight links
* update model-index.yml
* fix lint
* fix typo
* Update efficientnetv2-b1_8xb32_in1k.py
* Update efficientnetv2-b2_8xb32_in1k.py
* Update efficientnetv2-b3_8xb32_in1k.py
* update two moduals and model file
* update modual file
* update accuracys
* update accuracys
* update metafile
* fix build docs
* update links
* update README.md
Co-authored-by: qingtian <459291290@qq.com>
Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
2022-12-30 15:18:39 +08:00
9038c1c255
[Feature] Support TTA and add `--tta` in `tools/test.py`. ( #1161 )
...
* [Feature] Support TTA and add `--tta` in `tools/test.py`.
* Add unit tests.
* Rename the TTA model to `AverageClsScoreTTA`.
2022-12-30 11:46:17 +08:00
5b266d9e7c
[Feature] Add clip backbone. ( #1258 )
...
* clip backbone added
* passed precommit
* update readme
* update according to PR review
* add missing file
* add unittest
* refine metafile
* refine metafile and readme for readdocs
* refine metafile
* refine metafile
* Update metafile
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-12-22 16:33:57 +08:00
14dcb69092
[Feature] Add mixmim backbone with checkpoints. ( #1224 )
...
* add mixmim backbone
* add mixmim inference
* add docstring, metafile, test and modify readme
* Update README and metafile
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-12-20 16:52:54 +08:00
5547f4cac4
[Feature] Add TinyViT for dev-1.x. ( #1042 )
...
* [Feature] add TinyViT for dev-1.x
* [Feature] update readme
* fix lint error
* refactor the code
* [Update] update the args
* [Update] add unit test and fix bugs
* Rename the configuration file
* delete invalid files
* [Feature] update tinyvit readme
* [Feature] update tinyvit readme
* [Feature] update metafile
* Update tinyvit metafile
2022-12-20 13:04:00 +08:00
9e82db6032
[Enhance] Support ConvNeXt More Weights. ( #1240 )
...
* convnext more weights
* Update metafile and README
* Fix link
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-12-19 17:17:57 +08:00
3006fa26ab
[Fix] Fix CAM visualization. ( #1248 )
2022-12-19 13:54:52 +08:00
2535c1ecd7
[Feature] Support EVA. ( #1239 )
...
* add eva
* add eva
* add eva
* sklearn -> scikit-learn
* add large
* Update model names and links.
* Fix resize pos embed error when loading fp16 weight.
* Remove verbose configs.
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-12-14 13:21:33 +08:00
210373c093
[Feature] Implementation of RevViT. ( #1127 )
...
* [Feature] implement rev-vit network
* can reproduce the RevViT-Small accuracy 79.9
* update
* [Feature] update revvit
* [Feature] update revvit readme
* Update links
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-12-14 11:46:39 +08:00
ef3610d962
[Bug] Fix `reparameterize_model.py` doesn't save meta info. ( #1221 )
...
* fix reparameterize_model.py don't save meta info
* fix error
* Fix symbol link
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-12-06 17:06:31 +08:00
4fb44f8770
[Feature] EfficientNets NoisyStudent & L2. ( #1122 )
...
* add mvit 21k
* add mvit 21k
* add effnet noisy student
* Revert "add mvit 21k"
This reverts commit f51067c559
2022-11-21 16:57:13 +08:00
743ca2d602
[Fix] Fix the torchserve. ( #1143 )
...
* rebase
* update docker and rm deprecated deployment tools
* update docs
* rebase
Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
2022-11-21 11:18:42 +08:00
4969830c8a
[Enhance] Reproduce mobileone training accuracy. ( #1191 )
...
* add switch hook and UTs
* update doc
* update doc
* fix lint
* fix ci
* fix ci
* fix typo
* fix ci
* update configs names
* update configs
* update configs
* update links
* update readme
* update vis_scheduler
* update metafile
* update configs
* rebase
* fix ci
* rebase
2022-11-21 10:43:34 +08:00
72c6bc4864
[Feature] Support RepLKnet backbone. ( #1129 )
...
* update replknet configs
* update replknet test
* update replknet model
* update replknet model
* update replknet model
* update replknet model
* Fix docs and config names
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-11-21 10:18:58 +08:00
f458bf5a64
[Docs] update visualization doc. ( #1160 )
...
* [Docs] update visualization doc
* update doc
* update folder
* update analysis
* Update print config tool
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-11-18 12:44:55 +08:00
c4f3883a22
[Feature] Support DaViT. ( #1105 )
...
* add davit
* fix mixup config
* convert scripts
* lint
* test
* test
* Add checkpoint links.
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-11-16 17:23:55 +08:00
ee9ee9cf3c
[Fix] Update mmcls2torchserve. ( #1189 )
2022-11-14 14:17:11 +08:00
c48cfa9f47
[Docs] Improve the ResNet model page. ( #1118 )
...
* [Docs] Improve the ResNet model page.
* Fix lint.
* Update stat.py
2022-11-08 11:09:51 +08:00
d05cbbcf9b
[Feature] Support HorNet Backbone for dev1.x. ( #1094 )
...
* add hornet
* add hornet
* fix mixup config
* add optim cfgs
Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
2022-11-04 15:33:46 +08:00
29c46c8af2
[Docs] Add runtime configuration docs. ( #1128 )
...
* [Docs] Add runtime configuration docs.
* Fix grammar errors.
* Imporve docs according to comments
2022-11-02 10:59:59 +08:00
9bc58745d1
[Enhance] Update analyze_results.py for dev-1.x. ( #1071 )
...
* update analyze_results
* lint
* add --rescale-factor and fix filename logic
* lint
2022-10-11 11:34:23 +08:00
a49c3076e1
[Feature] Support DeiT3. ( #1065 )
...
* deit3
deit3
lint
* add tools and test
* deit3
* deit3
* fix preprocess
* lint
* Update config names and checkpoint paths
* Update convert tools to use mmengine, and fix docstring.
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-10-10 14:54:20 +08:00
080eb79f94
[Fix] Fix verify dataset tool in 1.x ( #1062 )
2022-09-30 10:08:02 +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
789884bf08
[Refactor] Fix visualization tools. ( #1045 )
...
* update browse dataset
* update images
* update tools
* update grad_cam
* update docs
* update tools
* update docs
2022-09-20 18:09:05 +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
9999da646c
[Imporve] Update tools to enable `pin_memory` and `persistent_workers` by default. ( #1024 )
2022-09-14 11:57:32 +08:00
4367d05dfc
[Enhance] Use `fvcore` to calculate FLOPS. ( #1000 )
...
* [Feature] Use fvcore for flops count
* update requirements
* update
2022-08-31 11:04:42 +08:00
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
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
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
7f0502867b
[Improve] Add `--amp`, `--resume` and `--auto-scale-lr` options in `tools/train.py`. ( #914 )
2022-07-21 17:34:04 +08:00
1c82863449
fix lint
2022-07-18 11:11: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
24bcf069f8
[Refactor] Refactor dev scripts
2022-07-18 11:11:13 +08:00
d8f556668e
Refactor vis scheduler tool
2022-07-18 11:11:13 +08:00
5a3e736b4e
[Feature] Implement offline evaluation.
2022-07-18 11:11:13 +08:00
a82de04b67
Enhance browse datset
2022-07-18 11:11:13 +08:00
ee6e585e41
Mazerun/base classifier
2022-07-18 11:11:13 +08:00
522ab1fd84
[Feature] Add Visualization hook
2022-07-18 11:11:04 +08:00
4d4b22377d
Add browse dataset
2022-07-18 10:53:56 +08:00
Ma Zerun
aso538
szwlh-c
CSH
QINGTIAN
Rongjie Li
Wangbo Zhao(黑色枷锁)
Songyang Zhang
takuoko
kitecats
marouane amzil
Ezra-Yu
Jiahao Wang
Austin Welch
Ezra-Yu