df3d2d47f0
update resnext ckpts ( #283 )
2021-06-01 14:57:46 +08:00
5ff158cec9
[Bug] Fix Cifar color channel ( #280 )
...
* fix bug
* updatedoc accordingly
2021-05-31 14:15:22 +08:00
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
b23b319f56
fix version ( #276 )
2021-05-29 10:47:16 +08:00
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
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
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
37167158e7
bump version to v0.11.0 ( #233 )
2021-05-01 22:26:39 +08:00
f970a36898
update pytorch2onnx.md table ( #229 )
2021-04-30 17:18:06 +08:00
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
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
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
d72d233f9b
fix resnext resize ( #222 )
2021-04-25 16:26:34 +08:00
3f085026cf
[Feature] Add Cifar100 config ( #208 )
...
* add config for cifar100
* fix doc
2021-04-13 20:15:29 +08:00
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
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
1f6549eeee
bump version to 0.10.0 ( #194 )
2021-04-01 10:39:18 +08:00
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
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
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
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
c8109d1eca
Update stat to classify papers ( #139 )
...
* update stat to classify papers
* modify tag of papers
2021-01-14 16:48:05 +08:00
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
3a8d8a2455
remove the models with 0 ckpt and ignore the repeat papers when counting papers ( #135 )
2021-01-08 20:40:44 +08:00
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
d291f2d889
Refactor docs ( #126 )
...
* refactor docs
* revised according to comments
2020-12-28 19:47:06 +08:00
3e20b5efa6
add modelzoo statistics ( #119 )
2020-12-25 17:17:43 +08:00
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
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
374b13c419
replace urls in model zoo ( #102 )
2020-12-01 14:23:32 +08:00
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
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
b577594d3d
fix outdated configs
2020-10-01 16:15:14 +02:00
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
b6774b1224
Update imagenet data setup in getting_started document ( #48 )
2020-09-30 01:06:23 +08:00
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
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
8d3acce307
Add ResNeSt ( #25 )
...
* Add ResNeSt
* fixed test
* refactor
* add ResNeSt base
* update modelzoo
* update modelzoo
* Add S-200,S269 _base_
2020-09-22 10:41:51 +08:00
6b0b3da92a
Add note about ground-truth labels range ( #44 )
...
* Add note about ground-truth labels range
* Fix
2020-09-09 16:21:25 +08:00
4b46fd6dc7
Add CIFAR10 configs and models ( #38 )
...
* rename cifar configs
* add cifar10 configs
* fix linting
* add params, flops and accuracy in docs
* add cifar10 pretrained models and logs
* use oss-accelerate url
* del unused cifar10 configs
2020-08-26 16:34:15 +08:00
b2fc837e84
Rename configs with samples_per_batch x gpus ( #37 )
...
* rename configs with samples_per_batch x gpus
* revise docs accordingly
2020-08-23 23:53:51 +08:00
59abed3c39
Update pretrained model ( #33 )
...
* update model_zoo
* update performance
* update shufflenet v2
Co-authored-by: lixiaojie <lixiaojie@sensetime.com>
2020-08-13 14:04:57 +08:00
22632ae24c
correct module name from mmdet to mmcls ( #8 )
2020-07-21 19:28:14 +08:00
41fae9b66b
add docs conf
2020-07-13 00:47:13 +08:00
4e6875d44e
add index.rst for readthedocs
2020-07-12 00:06:57 +08:00
5c88675a71
Add configs for MNIST, CIFAR10 and ImageNet
2020-07-12 00:06:57 +08:00
563fa63d06
Add model_zoo docs
2020-07-12 00:06:57 +08:00
6968ad5b3b
Add RandomFlip
2020-07-12 00:06:56 +08:00
4bc2268f48
Add tutorial docs
2020-07-12 00:06:54 +08:00
85399b397d
Add installation docs
2020-07-12 00:06:53 +08:00
LXXXXR
Ma Zerun
QingChuanWS
whcao
mzr1996
ftbabi
congee
David de la Iglesia Castro
Lei Yang
Mansi Mane
Jerry Jiarui XU
Xiaojie Li
yl-1993
yanglei