6038df9514
Update docs.
2023-03-20 16:03:57 +08:00
f6b65fcbe7
[Docs] Update get start docs and user guides. ( #1407 )
...
* update user_guides
* update test.md
* fix lint
* fix typo
* refine
* fix typo
* update retriever to api
* update rst and downstream
* update index.rst
* update index.rst
* update custom.js
* update chinese docs
* update config.md
* update train and test
* add pretrain on custom dataset
* fix lint
2023-03-20 15:56:09 +08:00
6cedce234e
[Refactor] Update dev scripts to be compatible with selfsup tasks. ( #1412 )
...
* [Refactor] Update dev scripts to be compatible with selfsup tasks.
* Fix some missing fields in config files.
* Set maximum number of gpus for local training.
* Update README files
* Update according to comments.
2023-03-20 14:30:57 +08:00
8875e9da92
[Docs] Update migration.md ( #1417 )
...
* update migration
* refine table
* update zh_cn
* fix lint
* Polish the documentation by ChatGPT.
* Update sphinx version and fix some warning.
---------
Co-authored-by: mzr1996 <mzr1996@163.com>
2023-03-17 10:30:09 +08:00
3472ee5d2c
[Feature] Implememnt the universal visualizer for multiple task. ( #1404 )
...
* [Feature] Implememnt the universal visualizer for multiple task.
* Update tools
* Improve according to comments.
* Fix tools docs
* Add --test-cfg option and set default collate function.
2023-03-09 11:36:54 +08:00
274a67223e
[Feature] Implement layer-wise learning rate decay optimizer constructor. ( #1399 )
...
* [Feature] Implement layer-wise learning rate decay optimizer constructor.
* Use num_layers instead of max_depth to avoid misleading
* Add UT
* Update docstring
* Update log info
* update LearningRateDecay configs
---------
Co-authored-by: fangyixiao18 <fangyx18@hotmail.com>
2023-03-07 17:30:39 +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
a05c79e806
[Refactor] Move transforms in mmselfsup to mmpretrain. ( #1396 )
...
* [Refactor] Move transforms in mmselfsup to mmpretrain.
* Update transform docs and configs. And register some mmcv transforms in
mmpretrain.
* Fix missing transform wrapper.
* update selfsup transforms
* Fix UT
* Fix UT
* update gaussianblur inconfigs
---------
Co-authored-by: fangyixiao18 <fangyx18@hotmail.com>
2023-03-03 15:01:11 +08:00
e035e03d59
Update docs style.
2023-03-02 13:29:48 +08:00
dda3d6565b
[Docs] Update generate_readme.py and readme files. ( #1388 )
...
* Update generate_readme.py and readme files.
* Update reamde
* Update docs
* update metafile
* update simmim readme
* update
* update mae
* fix lint
* update mocov2
* update readme pic
* fix lint
* Fix mmcls download links.
* Fix Chinese docs.
* Decrease readthedocs requirements.
---------
Co-authored-by: fangyixiao18 <fangyx18@hotmail.com>
2023-03-02 13:29:07 +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
8352951f3d
[Feature] Support XCiT Backbone. ( #1305 )
...
* update model file
* Update XCiT implementation and configs.
* Update metafiles
* Update metafile
* Fix floor divide
* Imporve memory usage
---------
Co-authored-by: qingtian <459291290@qq.com>
Co-authored-by: mzr1996 <mzr1996@163.com>
2023-02-15 10:32:35 +08:00
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
841256b630
[Feature] Support RetrieverRecall metric & Add ArcFace config ( #1316 )
...
* rebase
* add ap metric
* fix mlti-gpu bug in retrevel
* rebase
* rebase
* add training cfgs and update readme.md
* fix bugs(cannot load vecs in dist and diff test-val recall\)
* update configs and readme
* fix ut
* fix doc
* rebase
* fix rebase conflicts
* fix rebase error
* fix UT error
* fix docs
* fix typo
---------
Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
2023-02-14 12:46:21 +08:00
1c1273abca
[Docs] Translate some tools tutorials to Chinese. ( #1321 )
...
* Delete print_config.md
* Add files via upload
2023-02-09 16:40:13 +08:00
7ec6062415
[Refactor] Unify the `--out` and `--dump` in `tools/test.py`. ( #1307 )
2023-02-09 14:05:03 +08:00
a3f2effb17
[Feature] Add `ImageClassificationInferencer`. ( #1261 )
...
* [Feature] Add ImageClassificationInferencer.
* Update inferencer implementation and add unit tests.
* Update documentations.
* Update pre-commit hook
* Update docs
2023-02-08 14:30:12 +08:00
7e4502b0ac
[Feature] Support InShop Dataset (Image Retrieval). ( #1019 )
...
* rebase
* feat: add inshop dataset (retrieval)
* update fileIO
* update unit tests
* fix windows ci
* fix windows ci
* fix lint
* update unit tests
* update docs
* update docs
Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
2023-01-18 17:16:54 +08:00
353886eaca
[Docs] Add Chinese translation for runtime.md. ( #1313 )
...
* [Docs] Add Chinese translation for runtime.md
* [Docs] Update Chinese translation for runtime.md
* [Docs] Update for runtime.md
* [Docs] Update for runtime.md
* [Docs] Update for runtime.md
2023-01-18 17:15:43 +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
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
0d8f918eaa
Bump version to v1.0.0rc5. ( #1286 )
2022-12-30 17:32:04 +08:00
4f5350f365
[Doc] Fix typo. ( #1281 )
...
* [Fix] Fix imports in transforms. (#1255 )
* fix import
* import from mmegine.utils
* 修复错别字
Co-authored-by: Xieyuan Zhang <25652281+Francis777@users.noreply.github.com>
2022-12-30 15:52:57 +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
b63515111b
[Reproduce] Update ConvNeXt config files. ( #1256 )
...
* Update ConvNeXt training configs.
* Update ConvNeXt network.
* Update metafile and README.
* Update README
2022-12-19 13:54:24 +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
458ac4c89a
Merge remote-tracking branch 'origin/dev-1.x' into 1.x
2022-12-06 18:00:59 +08:00
12eca5b94a
Bump version to v1.0.0rc4. ( #1243 )
...
* Bump version to v1.0.0rc4
* Update according to comments
2022-12-06 18:00:01 +08:00
a4ec2799d6
[Docs] Update install tutorial. ( #1223 )
...
* [Docs] Update install tutorial.
* Update collapsed sections.
* Update requirements
* Update install tutorial.
2022-12-06 17:00:32 +08:00
c127c474b9
[Feature] Support getting model from the name defined in the model-index file. ( #1236 )
...
* [Feature] Support getting model from the name defined in the model-index file.
* Add unit tests.
* Prevent import `timm` if the `TIMMBackbone` is not used.
* Fix Windows CI.
* Move `init_model` to `mmcls.apis.hub`, and support pass nn.Module to all
model components.
* Fix requirements
* Rename `hub.py` to `model.py` and add unit tests.
2022-12-06 17:00:22 +08:00
7b9a1010f5
[Enhance] Support evaluate on both EMA and non-EMA models. ( #1204 )
...
* [Enhance] Support evaluate on both EMA and original models.
* Fix lint
2022-12-05 14:16:12 +08:00
d80ec5a4b8
[Refactor] Refactor BEiT backbone and support v1/v2 inference. ( #1144 )
...
* refactor beit backbone
* use LinearClsHead
* fix mean and std value
* fix lint
* support inference if beit-v2
* update encoder layer and init
* update
* add ut
* add prepare_relative_position_bias_table function
* add cls_token
* fix lint
* add pos_embed check
* update metafile and readme
* update weights link
* update link of weights
* update metafile
* update
* update docstrings
* update according to review
* rename readme
* update docstring
* fix lint
2022-11-29 12:56:33 +08:00
f9be21ab74
[Docs] Add version selection in the banner. ( #1217 )
...
* [Docs] Add version selection in the banner.
* Small fix.
2022-11-23 13:38:23 +08:00
75e502ed75
Update Chinese translation.
2022-11-22 15:19:03 +08:00
a4cfd55dd2
Update chinese translation.
2022-11-22 11:39:49 +08:00
44d2886422
[Docs] Add some Chinese translation for API pages.
2022-11-22 09:49:39 +08:00
13ff394985
Bump version to v1.0.0rc3. ( #1211 )
...
* Bump version to v1.0.0rc3
* Update pre-commit hook
2022-11-21 18:21:48 +08:00
b0007812d6
[Enhance] Enhance ArcFaceClsHead. ( #1181 )
...
* update arcface
* fix unit tests
* add adv-margins
add adv-margins
update arcface
* rebase
* update doc and fix ut
* rebase
* update code
* rebase
* use label data
* update set-margins
* Modify Arcface related method names.
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-11-21 18:10:39 +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
629f6447ef
[Feature] Migrate CSRA head to 1.x. ( #1177 )
...
* [Feat] add csra to 1x
* minor fix
* add voc metrics
* refine
* add unittest
* minor fix
* add more comments
* Fix docs and metafile.
* Fix docs.
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-11-21 10:39:16 +08:00
0e8cfa6286
[Docs] Add not-found page extension. ( #1207 )
...
* [Docs] Add not-found page extension.
* Mock rich during generate docs.
* Fix multiple broken links in docs.
* Fix "right" to "left".
2022-11-21 10:34:05 +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
c3c1cb93aa
[Feature] Add Switch Recipe Hook. ( #1101 )
...
* add switch hook and UTs
* update doc
* update doc
* fix lint
* fix ci
* fix ci
* fix typo
* fix ci
* switchTrainAugHook to switchRecipeHook
* fix lint
* Refactor the `SwitchRecipeHook`.
* Fix windows CI
* Fix windows CI
* Fix windows CI.
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-11-18 18:12:03 +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
542143cb41
[Feature] Add TIMM and HuggingFace wrappers to build classifiers from them directly. ( #1102 )
...
* [Feature] Add TIMM and HuggingFace wrappers to build classifiers from them directly.
* Support `with_cp` and add docstring.
* Add unit tests.
* Update CI.
* Update docs.
2022-11-10 14:56:19 +08:00
mzr1996
Yixiao Fang
QINGTIAN
takuoko
huangxuming
zzc98
GAO SHIQI
aso538
szwlh-c
Mr.Li
Songyang Zhang
Ezra-Yu
marouane amzil
Hubert
Jiahao Wang