Commit Graph

59 Commits (dbf3df21a31ee5404184b3f1ba0588b6b6aaa8f4)

Author SHA1 Message Date
Ma Zerun ade7b80e44
Bump version to v0.14.0 (#389) 2021-08-04 13:25:42 +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
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
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 b95caca0d5
Redirect model_zoo.md and format Chinese documents. (#290)
* Redirect model_zoo.md in Chinese Readme to English ver.

* Format Chinese documents.
2021-06-03 16:49:53 +08:00
LXXXXR 27a49a9646
bump version to v0.12.0 (#287) 2021-06-03 11:42:34 +08:00
LXXXXR dac090162d
Bump version to v0.11.1 (#256)
* bump version to v0.11.1

* minor fix

* minor fix

* minor fix

* minor fix
2021-05-21 16:36:08 +08:00
LXXXXR 37167158e7
bump version to v0.11.0 (#233) 2021-05-01 22:26:39 +08:00
mzr1996 3d87474c71
Add Chinese Readme and docs (#221)
* Add Citation in README

Add Citation and mmgeneration in README

* Merge inference and test section in getting_start.md and other small chagne.

* Fix code type in install.md

* Add Chinese Readme

* README and docs improvement.
2021-04-26 13:58:18 +08:00