Commit Graph

64 Commits (c92eefbb4a767d6bda1340daadb16d79ac2d00ef)

Author SHA1 Message Date
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
Ma Zerun 9352fc434a
Fix vgg-19 model link. (#363) 2021-07-22 11:29:10 +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 2ccc55ce4f
bump version to v0.13.0 (#340) 2021-07-05 10:19:57 +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
AllentDan a24a9f6faa
[Fix] Build compatible with low pytorch versions (#301)
* add version compatible for torchscript

* doc

* doc again

* fix lint

* fix lint isort
2021-06-14 23:25:35 +08:00
AllentDan c2f01e0dcd
[Feature] Add torchscript deployment (#279)
* add torchscript deploy

* fix lint

* add check and delete \
2021-06-12 21:50:48 +08:00
q.yao dbddde52ef
[Feature] TensorRT test tools. (#284)
* first commit

* update resnext result

* update docs

* update docstring
2021-06-12 21:47:10 +08:00
David de la Iglesia Castro e2507cae97
Add mmcls2torchserve (#292)
* Add mmcls2torchserve

* Update docs
2021-06-12 21:45:45 +08:00
LXXXXR c68bf9f2ba
fix resnetv1d configs and update ckpts (#288) 2021-06-03 16:30:29 +08:00
LXXXXR 27a49a9646
bump version to v0.12.0 (#287) 2021-06-03 11:42:34 +08:00
LXXXXR df3d2d47f0
update resnext ckpts (#283) 2021-06-01 14:57:46 +08:00
LXXXXR 5ff158cec9
[Bug] Fix Cifar color channel (#280)
* fix bug

* updatedoc accordingly
2021-05-31 14:15:22 +08:00
Ma Zerun 4524327a4a
[Docs] Fix typo, improve and add Chinese translation of `data_pipeline.md` and `new_modules.md` (#265)
* Improve docs, including `new_modules.md` and `data_pipeline.md`

* Add Chinese docs `new_modules.md` and `data_pipeline.md`
2021-05-29 10:48:31 +08:00
LXXXXR b23b319f56
fix version (#276) 2021-05-29 10:47:16 +08:00
Ma Zerun a06827bc08
Build chinese translation in readthedocs (#267)
* change document layout

* Add Chinese docs build file

* Add switch language option
2021-05-29 10:00:37 +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
mzr1996 9e93feabac
[Docs] Fix error in new_dataset.md and add Chinese translation of finture.md, new_dataset.md (#243)
* Fix error in new_dataset.md

* Add Chinese Translation of finture.md, new_dataset.md
2021-05-10 17:17:37 +08:00
LXXXXR 37167158e7
bump version to v0.11.0 (#233) 2021-05-01 22:26:39 +08:00
QingChuanWS f970a36898
update pytorch2onnx.md table (#229) 2021-04-30 17:18:06 +08:00
whcao 3be95b99c2
[Feature]Modify modelzoo readme (#230)
* add mytrain.py for test

* test before layers

* test attr in layers

* test classifier

* delete mytrain.py

* add imagenet_bs4096_AdamW.py

* delete 2 lines of comments

* change bs to 64

* fix bug

* add vit to model_zoo.md

* rename
2021-04-29 15:18:55 +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
QingChuanWS 01d2849b76
[Feature]: add onnxruntime test tool (#212)
* [draft] add onnxruntime accuruacy verification

* fix a bug

* update code

* fix lint

* fix lint

* update code and doc

* update doc

* update code

* update code

* updata doc and updata code

* update doc and fix some bug

* update doc

* update doc

* update doc

* update doc

* update doc

* update doc

* fix bug

* update doc

* update code

* move CUDAExecutionProvider to first place

* update resnext accuracy

* update doc

Co-authored-by: maningsheng <maningsheng@sensetime.com>
2021-04-26 13:57:08 +08:00
LXXXXR d72d233f9b
fix resnext resize (#222) 2021-04-25 16:26:34 +08:00
LXXXXR 3f085026cf
[Feature] Add Cifar100 config (#208)
* add config for cifar100

* fix doc
2021-04-13 20:15:29 +08:00
QingChuanWS 2bd28435cf
Add simplify option in pytorch2onnx.py and Update doc (#200)
* add simplify option in pytorch2onnx.py and update relevant doc

* update doc about pytorch2onnx

* update code

* update docs
2021-04-09 13:05:25 +08:00
QingChuanWS d259306013
Add pytorch to onnx and onnx to tensorrt doc (#180)
* add pytorch2onnx and onnx2trt doc

* fix the typo in deploy doc

* fix bug of onnx2trt.md

* Remove redundant install steps in onnx2trt.md

* update doc based on code review

* update index.rst
2021-04-01 15:56:55 +08:00
LXXXXR 1f6549eeee
bump version to 0.10.0 (#194) 2021-04-01 10:39:18 +08:00
ftbabi 49ece048e0
Add tutorials (#176)
* Add tutorials using MobileNetV2

* Modify text and use ResNet50

Co-authored-by: wangruohui <12756472+wangruohui@users.noreply.github.com>
2021-03-17 16:29:21 +08:00
LXXXXR 7ca0ca271f
Bump version to 0.9.0 (#166)
* update docs for version 0.9.0

* add missing #153
2021-03-01 20:14:53 +08:00
LXXXXR 2d7dc997e6
[Fix] Fix vgg configs and upload checkpoints (#161)
* fix configs for vgg

* change lr

* rename configs for vgg

* fix vgg in model zoo
2021-02-20 19:37:38 +08:00
LXXXXR 7f49632d7c
Bump version to 0.8.0 (#156)
* update docs to bump version

* minor change

* revise acording to comments
2021-01-31 17:50:40 +08:00
congee c8109d1eca
Update stat to classify papers (#139)
* update stat to classify papers

* modify tag of papers
2021-01-14 16:48:05 +08:00
LXXXXR 194ab7efda
[Feature] Add bce loss for multilabel task (#130)
* add bce loss for multilabel task

* minor change

* apply class wise sum

* fix docstring

* do not apply sum over classes and fix docstring

* fix docstring

* fix weight shape

* fix weight shape

* fix docstring

* fix linting issue

Co-authored-by: Y. Xiong <xiongyuxy@gmail.com>
2021-01-11 11:05:24 +08:00
congee 3a8d8a2455
remove the models with 0 ckpt and ignore the repeat papers when counting papers (#135) 2021-01-08 20:40:44 +08:00
LXXXXR d835cd005a
Bump version to 0.7.0 (#129)
* add change log and unable MD024 in pre-commit hook

* fix readme and modify version

* Revised according to comments
2020-12-31 16:41:45 +08:00
LXXXXR d291f2d889
Refactor docs (#126)
* refactor docs

* revised according to comments
2020-12-28 19:47:06 +08:00
congee 3e20b5efa6
add modelzoo statistics (#119) 2020-12-25 17:17:43 +08:00
LXXXXR 736eec1fd2
Fix links in docs (#118)
* fix links in docs

* Revised according to comments

* revised according to comments
2020-12-21 15:29:12 +08:00
LXXXXR 400074c321
Add markdown lint in pre-commit hook (#106)
* add mdlint

* remove duplicate check

* modify md files to pass linting check

* change mmcv download url

* remove torch1.4.0 version check

* remove torch1.4.0 version check
2020-12-02 19:42:45 +08:00
LXXXXR 374b13c419
replace urls in model zoo (#102) 2020-12-01 14:23:32 +08:00
LXXXXR 761c39b6b9
Add config link in model_zoo.md (#99)
* add config link in model_zoo.md

* change model download path
2020-11-26 15:47:22 +08:00
David de la Iglesia Castro 4f4f2957ef
Use build_runner (#54)
* Use build_runner in train api

* Support iter in eval_hook

* Add runner section

* Add test_eval_hook

* Pin mmcv version in install docs

* Replace max_iters with max_epochs

* Set by_epoch=True as default

* Remove trailing space

* Replace DeprecationWarning with UserWarning

* pre-commit

* Fix tests
2020-10-15 21:12:50 +08:00
David de la Iglesia Castro b577594d3d fix outdated configs 2020-10-01 16:15:14 +02:00
Lei Yang 9547e7b7a5
Add model inference (#16)
* add model inference on single image

* rm --eval

* revise doc

* add inference tool and demo

* fix linting

* rename inference_image to inference_model

* infer pred_label and pred_score

* fix linting

* add docstr for inference

* add remove_keys

* add doc for inference

* dump results rather than outputs

* add class_names

* add related infer scripts

* add demo image and the first part of colab tutorial

* conduct evaluation in dataset

* return lst in simple_test

* compuate topk accuracy with numpy

* return outputs in test api

* merge inference and evaluation tool

* fix typo

* rm gt_labels in test conifg

* get gt_labels during evaluation

* sperate the ipython notebook to another PR

* return tensor for onnx_export

* detach var in simple_test

* rm inference script

* rm inference script

* construct data dict to replace LoadImage

* print first predicted result if args.out is None

* modify test_pipeline in inference

* refactor class_names of imagenet

* set class_to_idx as a property in base dataset

* output pred_class during inference

* remove unused docstr
2020-09-30 19:00:20 +08:00
Mansi Mane b6774b1224
Update imagenet data setup in getting_started document (#48) 2020-09-30 01:06:23 +08:00
Lei Yang bc1b08ba41
Add VGG and pretained models (#27)
* add vgg

* add vgg model coversion tool

* fix out_indices and docstr

* add vgg models in configs

* add params, flops and accuracy in docs

* add pretrained models url

* use ConvModule and refine var names

* update vgg conversion tool

* modify bn config

* add docs for arch_setting

* add unit test for vgg

* rm debug code

* update vgg pretrained models
2020-09-29 17:49:42 +08:00
David de la Iglesia Castro 99115fddbc
Add albumentations (#45)
* Add Albu transform

* pre-commit

* Create optional.txt

* Update requirements.txt

* Update transforms.py
2020-09-22 17:35:39 +08:00