Commit Graph

460 Commits (f9a2b04cee1cd6a416b6b2731e7112c12520dda5)
 

Author SHA1 Message Date
Ma Zerun 5aa132a91c
[CI] Add PyTorch 1.9 and Python 3.9 build workflow, and remove some CI. (#422)
* Add PyTorch 1.9 build workflow, and remove some CI.

* Add Python 3.9 CI

* Show Python 3.9 support.
2021-09-01 11:06:00 +08:00
Ma Zerun a41cb2fa93
Bump version to v0.15.0 (#426) 2021-08-31 14:33:32 +08:00
Ma Zerun 5383787512
[Enhance] Better result visualization (#419)
* Imporve result visualization to support wait time and change the backend
to matplotlib.

* Add unit test for visualization

* Add adaptive dpi function

* Rename `imshow_cls_result` to `imshow_infos`.

* Support str in `imshow_infos`

* Improve docstring.
2021-08-31 10:50:28 +08:00
Ezra-Yu 192b79eea0
[Feature] Support classwise weight in losses (#388)
* Add classwise weight in losses:CE,BCE,softBCE

* Update unit test

* rm some extra code

* rm some extra code

* fix broadcast

* fix broadcast

* update unit tests

* use new_tensor

* fix lint
2021-08-31 10:44:12 +08:00
Ma Zerun 6a0a76af0c
[Enhance] Add `hparams` argument in `AutoAugment` and `RandAugment` and some other improvement. (#398)
* Add hparams argument in `AutoAugment` and `RandAugment`.

And `pad_val` supports sequence instead of tuple only.

* Add unit tests for `AutoAugment` and `hparams` in `RandAugment`.

* Use smaller test image to speed up uni tests.

* Use hparams to simplify RandAugment config in swin-transformer.

* Rename augment config name from `pipeline` to `pipelines`.

* Add some commnet ad docstring.
2021-08-24 18:15:54 +08:00
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