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

29 Commits (4d1fb1a662f8285a7b6850ab64861ff4c8c3887d)

Author SHA1 Message Date
LXXXXR 77e88844cd
[Bug] Bug in vgg config (#181) 
* fix vgg_voc config

* fix vgg_voc config
 2021-03-25 15:50:01 +08:00
LXXXXR d3dceddf31
add model zoo for fp16 (#184) 2021-03-25 15:49:26 +08:00
LXXXXR e76c5a368d
[Feature] Support fp16 training (#178) 
* change mmcls fp16 to mmcv hook

* support fp16

* clean unnessary stuff
 2021-03-17 15:53:55 +08:00
ftbabi bdd6b01ae7
[Feature] Add "mixup" from Bag of Tricks (#160) 
* Add mixup option

* Modify the structure of mixup and add configs

* Clean configs

* Add test for mixup and SoftCrossEntropyLoss

* Add simple test for ImageClassifier

* Fix bug in test_losses.py

* Add assertion in CrossEntropyLoss
 2021-02-25 14:06:58 +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 07bb15e5fd
[Feature] Add heads and config for multilabel task (#145) 
* resolve conflicts
add heads and config for multilabel tasks

* minor change

* remove evaluating mAP in head

* add baseline config

* add configs

* reserve only one config

* minor change

* fix minor bug

* minor change

* minor change

* add unittests and fix docstrings
 2021-01-25 18:10:14 +08:00
Cheng-Yen Yang 87d4a1fcea
Update README.md (#150) 2021-01-25 17:16:57 +08:00
LXXXXR aa50103442
fix typo in configs (#147) 2021-01-21 14:47:02 +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 d062cdef0a
[Docs] Add tags in README.md (#137) 
* add tag in README.md

* fix linting issue
 2021-01-11 11:10:06 +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 668292f5ab
Add README.md in configs (#117) 
* rearrange config files

* rearrange config files

* add README.md in configs

* rename
 2020-12-18 20:04:54 +08:00
LXXXXR 384ff95479
Rearrange config files (#116) 
* rearrange config files

* rearrange config files

* add README.md in configs

* Revert "add README.md in configs"

This reverts commit 68ff996f91.

* rename
 2020-12-18 16:59:50 +08:00
WRH 44bbc71e14
Fix bug and optimize MNIST config (#98) 
* add simple_test to ClsHead

* optimize lenet training config

* recover path setting
 2020-11-26 15:27:04 +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
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
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
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 18ac0f0fa3
Update shufflenet config (#31) 
* update shufflenet config

* remove old config

Co-authored-by: lixiaojie <lixiaojie@sensetime.com>
 2020-08-13 10:46:39 +08:00
Xiaojie Li 414ede4ca4
fix coslr (#32) 
Co-authored-by: lixiaojie <lixiaojie@sensetime.com>
 2020-08-13 10:45:41 +08:00
Xiaojie Li 608cea2469
fix (#30) 
Co-authored-by: lixiaojie <lixiaojie@sensetime.com>
 2020-08-13 09:50:14 +08:00
yl-1993 7d1edd03c6 revise a part of configs to pil resize 2020-08-03 12:54:58 +08:00
yl-1993 8af1537569 add pil resize configs 2020-08-03 12:54:08 +08:00
yl-1993 71e3b86ed3 add github action (manually squash) 2020-07-12 00:07:05 +08:00
yanglei 5c88675a71 Add configs for MNIST, CIFAR10 and ImageNet 2020-07-12 00:06:57 +08:00
chenkai a99e6eb10d init commit 2020-05-21 21:21:43 +08:00