Commit Graph

355 Commits (a9d65271abfcb3137b3defc9fa03049615fe51da)
 

Author SHA1 Message Date
Ma Zerun a9d65271ab
[Docs] Add algorithm readme and update meta yml (#418)
* Add README.md for models without checkpoints.

* Update model-index.yml

* Update metafile.yml of seresnet
2021-08-24 17:46:46 +08:00
Ezra-Yu 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
Ma Zerun f9eb9b409b
[Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
Azad Baykara 5c4da16084
[Docs] Fix typo 'metirc' (#411) 2021-08-17 10:35:26 +08:00
Ma Zerun c0a4916a06
[Enhance] Update `digit_version` function. (#402)
* Update digit_version

* Add digit_version unit test
2021-08-12 12:13:25 +08:00
Ma Zerun f8f1700860
[Refactor] Use post_process function to handle pred result processing. (#390)
Use post_process function to handle pred result processing in `simple_test`.
2021-08-12 11:54:24 +08:00
Ma Zerun e4188c828c
Avoid albumentations to install both opencv and opencv-headless. (#397) 2021-08-09 10:32:13 +08:00
Vladislav Sovrasov b54acfd5c4
[Enhance] Avoid unnecessary listdir when building ImageNet. (#396) 2021-08-06 10:38:36 +08:00
Ma Zerun a747458ba8
[Docs] Update QQ group QR code. (#393) 2021-08-05 17:51:53 +08:00
Miguel Méndez 64bbed41f4
[Enhance] Use dynamic mmcv download link in TorchServe dockerfile. (#387)
* Add mmcv arg

* Declare ARG after FROM

Co-authored-by: Ma Zerun <mzr1996@163.com>
2021-08-04 17:46:40 +08:00
Ma Zerun 4aec976c87
Delete custom.md 2021-08-04 14:31:53 +08:00
Ma Zerun 2aa4100595
[Docs] Add PR template and modify issue template (#380)
* Add language suggestion in issue-template

* Add issue template config

* Add PR template

* Update pr template according to mmcv #1228
2021-08-04 14:30:20 +08:00
Ma Zerun ade7b80e44
Bump version to v0.14.0 (#389) 2021-08-04 13:25:42 +08:00
Ma Zerun df5cec3e92
[Docs] Fix readthedocs config lint (#384) 2021-07-31 16:20:42 +08:00
Ma Zerun e8822ebaec
[Docs] Fix some bugs in readthedocs pdf build. (#383)
* Add escape symbols for underscores in formulas.

* Support Chinese characters in pdf output.
2021-07-31 14:49:36 +08:00
Ma Zerun 469d88c21b
Support output pdf version readthedocs document. (#382) 2021-07-31 14:11:50 +08:00
whcao 359f56ad58
[Feature] Add transformer in transformer (#339)
* add tnt_small configs

* add tnt backbone

* test tnt

* add tnt to model_zoo

* rename the config file name

* add optimizor

* move tnt backbone unitest

* add metric

* fix keyname in arch

* encapsulate "inner transformer block" and "outer transformer block"

* fix TnT

* Use `inner_block_cfg` and `outer_block_cfg` instead of `args` and
`kwargs`.

Co-authored-by: mzr1996 <mzr1996@163.com>
2021-07-30 09:16:27 +08:00
pvys db856df43e
[Fix] Fix ImageNet dataset annotation file parse bug. (#370)
Co-authored-by: mzr1996 <mzr1996@163.com>
2021-07-29 10:21:22 +08:00
Ma Zerun c92eefbb4a
[Docs] Fix `Open in colab` link in Colab tutorials. (#377) 2021-07-27 19:57:21 +08:00
Junjun2016 e3b77e7508
[Fix] Fix `patch_cfg` argument bug in SwinTransformer (#368) 2021-07-27 19:19:44 +08:00
Junjun2016 1585ca83c8
[Fix] Fix docstring typo and init error in ShuffleNetV1 (#374) 2021-07-27 19:17:42 +08:00
Ma Zerun 33a4c0fe9f
[Fix] Use local ATTENTION registery to avoid conflict with other repo. (#375) 2021-07-27 17:59:05 +08:00
Ma Zerun 0f4cae4d3f
[Feature] Support mim (#376)
* Support mim extention

* Fix lint.

* Remove extra `demo` setting in mim related files.
2021-07-27 17:58:04 +08:00
Ezra-Yu 99e3d4e49e
[Feature] Add Dockerfile (#365)
* add Dockerfile

* Update Dockerfile

* Update Dockerfile

* Add docker install guide

* Add docker guide

* Update readme

* Update readme
2021-07-27 14:35:49 +08:00
WRH 19ff0d198c
[Docs] Add chinese ipynb tutorial (#306)
* add chinese ipynb

* Update Chinese Colab tutorials.

* Update colab docs to ensure deterministic.

* Add `open in colab` button.

Co-authored-by: mzr1996 <mzr1996@163.com>
2021-07-27 13:58:27 +08:00
Ma Zerun 899047a3b3
Fix duplicate `init_weights` call in ViT init function. (#373) 2021-07-26 05:33:11 -04:00
Ma Zerun 57e9692765
Fix broken _base_ link in a resnet config. (#361) 2021-07-25 23:50:32 -04:00
Ma Zerun 474c55d035
Fix mmcv dep version in mminstall.txt (#346) 2021-07-23 16:09:55 +08:00
Ma Zerun 9352fc434a
Fix vgg-19 model link. (#363) 2021-07-22 11:29:10 +08:00
Ma Zerun 15cd34bbef
[Fix] Use zero as default value of `thrs` in metrics. (#341)
* Use zero as default value of `thrs` in metrics. And it accepcts a number
instead of float now.

* Fix unit test comment

* Don't pass thrs if no thrs.
2021-07-18 16:57:21 +08:00
LXXXXR 24e58ba26d
[Feature] Add ResNeSt configs (#332)
* add configs for resnest training

* fix format
2021-07-14 15:58:19 +08:00
Ma Zerun 76c5d34dcc
[Feature] Support two options in `build_dataloader`. (#349)
* Support presistent_works in dataloader.

* Use pin_memory by default
2021-07-14 15:21:49 +08:00
Ma Zerun eb7c70c5c7
Create issue templates (#358)
* Create issue templates

* Fix lint.
2021-07-14 15:18:43 +08:00
Ma Zerun a97ccd5579
[Fix] Fix swin transformer config (#355)
* Fix config bug in swin

* Format config and checkpoint name of swin transformer.

* Fix link in model zoo
2021-07-14 15:10:20 +08:00
Ma Zerun d04ebc1eb5
[Docs] Add API Reference in the docs (#342)
* Add API inference in the docs and fix readthedocs config.

* Replace some relative link in docs.

* Format docstring for reStructuredText syntax.

* Fix vit paper link

* Fix docstring of `show_results` function in `BaseClassifier`.
2021-07-14 15:06:50 +08:00
Ma Zerun 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
Ma Zerun 71621a5f62
Add `is_tracing` helper function to fix a tracing bug in PyTorch 1.6 (#347) 2021-07-07 11:55:53 +08:00
Wenwei Zhang 06f415c28a
support to test mmdet inference with mmcls backbone (#343)
* support to test mmdet inference with mmcls backbone

* update config

* update

* add mmdet dependency in tests
2021-07-06 14:34:12 +08:00
Ma Zerun 2ccc55ce4f
bump version to v0.13.0 (#340) 2021-07-05 10:19:57 +08:00
whcao f2d0805496
replace model_zoo.yml with model-index.yml (#338) 2021-07-02 19:51:01 +08:00
whcao 5a86288f1f
[Fix]Fix metafile (#335)
* add mytrain.py for test

* test before layers

* test attr in layers

* test classifier

* delete mytrain.py

* add mobilenet_v3.yml

* fix ymal

* rename model_zoo.yml to model-index.yml

* replace imagenet with imagenet-1k
2021-07-01 20:50:42 +08:00
Ma Zerun 076ee10cac
[Feature] Add swin-transformer model. (#271)
* Add swin transformer archs S, B and L.

* Add SwinTransformer configs

* Add train config files of swin.

* Align init method with original code

* Use nn.Unfold to merge patch

* Change all ConfigDict to dict

* Add init_cfg for all subclasses of BaseModule.

* Use mmcv version init function

* Add Swin README

* Use safer cfg copy method

* Improve docstring and variable name.

* Fix some difference in randaug

Fix BGR bug, align scheduler config.

Fix label smoothing parameter difference.

* Fix missing droppath in attn

* Fix bug of relative posititon table if window width is not equal to
height.

* Make `PatchMerging` more general, support kernel, stride, padding and
dilation.

* Rename `residual` to `identity` in attention and FFN.

* Add `auto_pad` option to auto pad feature map

* Improve docstring.

* Fix bug in ShiftWMSA padding.

* Remove unused `key` and `value` in ShiftWMSA

* Move `PatchMerging` into utils and use common `PatchEmbed`.

* Use latest `LinearClsHead`, train augments and label smooth settings.
And remove original `SwinLinearClsHead`.

* Mark some configs as "Evalution Only".

* Remove useless comment in config

* 1. Move ShiftWindowMSA and WindowMSA to `utils/attention.py`
2. Add docstrings of each module.
3. Fix some variables' names.
4. Other small improvement.

* Add unit tests of swin-transformer and patchmerging.

* Fix some bugs in unit tests.

* Fix bug of rel_position_index if window is not square.

* Make WindowMSA implicit, and add unit tests.

* Add metafile.yml, update readme and model_zoo.
2021-07-01 09:30:42 +08:00
Mingqiang Ning 4ebee155e8
fix a bug when samples_per_gpu==1 (#311) 2021-06-30 20:57:21 +08:00
Ma Zerun 42b99be9b2
[Docs] rearrange docs and add multiple translation docs. (#320)
* Move tools docs to `tools` folder.

* Fix link error in model_serving.md

* Fix typo in CONTRIBUTING.md

* Add Chinese translation of CONTRIBUTING.md

* Add translation of `onnx2tensorrt.md`, `pytorch2onnx.md`,
`model_serving.md` and `pytorch2torchscript.md`.

* Improve tools docs.

* Add docs about installing mmcls via mim.
2021-06-30 20:53:09 +08:00
Ma Zerun aad796ae6f
Modify test tools and add some new tools (#322)
* Refactor tools folder structure.

* Modify tools/test.py and add eval_metric.py to analysis test output.

* Add new tools `analyze_logs.py` and `print_config.py`.

* Add comment for analysis_tools functions.
2021-06-30 20:51:16 +08:00
whcao bee0ac6b56
[Refactor]Modify patchembed (#330)
* add mytrain.py for test

* test before layers

* test attr in layers

* test classifier

* delete mytrain.py

* add patchembed and hybridembed

* add patchembed and hybridembed to __init__

* test patchembed and hybridembed

* fix some comments
2021-06-30 20:48:04 +08:00
Ezra-Yu 06bbd6940d
[Fix] Fix init_weights bug in some backbones and update Readme (#318)
* some init_weights bugs

* init_weights

* add import Basemodule

* code stye

* isort

* Use recommend init_weights override method.

Add init_cfg parameter in InvertedResidual.

Remove some useless comment.

Co-authored-by: mzr1996 <mzr1996@163.com>
2021-06-30 19:41:58 +08:00
Ma Zerun 19cfb25e5e
Use parameter default value to control default behavior of init_cfg in (#319)
`LinearClsHead` and `MultiLabelLinearClsHead`

And remove the verbose `_init_layers` method of `LinearClsHead` and
`MultiLabelLinearClsHead`.
2021-06-30 19:13:27 +08:00
Ma Zerun db5502f525
Fix config link of SeResNet README. (#326) 2021-06-29 23:19:58 +08:00
Ma Zerun 65410b05ad
Fix Mobilenetv3 structure and add pretrained model (#291)
* Refactor Mobilenetv3 structure and add ConvClsHead.

* Change model's name from 'MobileNetv3' to 'MobileNetV3'

* Modify configs for MobileNetV3 on CIFAR10.

And add MobileNetV3 configs for imagenet

* Fix activate setting bugs in MobileNetV3.

And remove bias in SELayer.

* Modify unittest

* Remove useless config and file.

* Fix mobilenetv3-large arch setting

* Add dropout option in ConvClsHead

* Fix MobilenetV3 structure according to torchvision version.

1. Remove with_expand_conv option in InvertedResidual, it should be decided by channels.

2. Revert activation function, should before SE layer.

* Format code.

* Rename MobilenetV3 arch "big" to "large".

* Add mobilenetv3_small torchvision training recipe

* Modify default `out_indices` of MobilenetV3, now it will change
according to `arch` if not specified.

* Add MobilenetV3 large config.

* Add mobilenetv3 README

* Modify InvertedResidual unit test.

* Refactor ConvClsHead to StackedLinearClsHead, and add unit tests.

* Add unit test for `simple_test` of `StackedLinearClsHead`.

* Fix typo

Co-authored-by: Yidi Shao <ydshao@smail.nju.edu.cn>
2021-06-27 23:19:36 +08:00