mirrors
/
mmpretrain
mirror of https://github.com/open-mmlab/mmpretrain.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
800 Commits
8 Branches
41 Tags
32 MiB
Commit Graph

236 Commits (8875e9da92de455c6770a87c245cbfad9ebcab42)

Author SHA1 Message Date
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
Jerry Jiarui XU 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
David de la Iglesia Castro 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
Lei Yang 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
Lei Yang 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
Xiaojie Li 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
Lei Yang 22632ae24c
correct module name from mmdet to mmcls (#8) 2020-07-21 19:28:14 +08:00
yl-1993 41fae9b66b add docs conf 2020-07-13 00:47:13 +08:00
yl-1993 4e6875d44e add index.rst for readthedocs 2020-07-12 00:06:57 +08:00
yanglei 5c88675a71 Add configs for MNIST, CIFAR10 and ImageNet 2020-07-12 00:06:57 +08:00
yanglei 563fa63d06 Add model_zoo docs 2020-07-12 00:06:57 +08:00
yanglei 6968ad5b3b Add RandomFlip 2020-07-12 00:06:56 +08:00
yanglei 4bc2268f48 Add tutorial docs 2020-07-12 00:06:54 +08:00
yanglei 85399b397d Add installation docs 2020-07-12 00:06:53 +08:00
yanglei d967b94439 Add getting_started docs 2020-07-12 00:06:53 +08:00