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

229 Commits (e76c5a368dc034d50553f8cd2b8d6136dda53078)
 

Author SHA1 Message Date
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
QingChuanWS 8eb845e718
support dynamic shape export to onnx (#175) 
* support dynamic shape export to onnx

* update the code and reset the simple_test() of image.py

* fix dynamic_axes name is not a valid input/output name
 2021-03-17 15:50:03 +08:00
LXXXXR f3b9380cbc
[Feature] Support solarize and posterize pipelines (#172) 
* support Solarize and Posterize

* revised according to comments

* fix conflicts
 2021-03-09 19:45:13 +08:00
LXXXXR 9614787fc4
[Feature] Add color pipeline (#171) 
* add ColorTransform pipeline

* fix docstring

* minor change

* revised according to comments
 2021-03-09 19:28:50 +08:00
LXXXXR c8033ece8e
add invert pipeline (#168) 2021-03-02 16:46:57 +08:00
LXXXXR b1fa298a66
add Rotate pipeline (#167) 2021-03-01 21:49:07 +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 bc50a7927b
reslove conflicts (#165) 2021-02-26 11:48:13 +08:00
LXXXXR 8c11c01fdb
[Feature] Add shear pipeline (#163) 
* half-done auto_augmentation

* remove auto_augcode and support shear pipeline

* fix typo

* fix typo

* use a non-square toy data instead
 2021-02-25 16:00:46 +08:00
David de la Iglesia Castro fb11a23cfe
Show results in test api (#162) 
* Initial support show in single_gpu_test

* Add result_show

* Add ori_filename meta_key

* test ori_filename

* Add show, show-dir, show-options
 2021-02-25 16:00:22 +08:00
LXXXXR 04cdbe4d66
remove unnecessary addopts (#164) 2021-02-25 14:09:03 +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
RunningLeon f0d358e340
add onnx2tensorrt (#153) 2021-02-05 17:48:37 +08:00
LXXXXR ddc2a14177
[Feature] Add --eval-options in test.py (#158) 
* add --eval-options in test.py

* fix typo

* revise according to commnets
 2021-02-05 17:46:43 +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
LXXXXR a7d4739d2b
[Fix] Fix test.py (#155) 
* modeify test.py

* modeify test.py

* revise according to comments
 2021-01-31 15:53:43 +08:00
LXXXXR a225cb6bdb
fix img_metas bug (#152) 2021-01-26 11:24:08 +08:00
LXXXXR 3e5a9513be
[Bug] Fix bug in Collect (#149) 
* fix bug in Collect

* add metakeys
 2021-01-25 20:29:28 +08:00
LXXXXR fc82c31b2f
fix RandomCrop and RandomResizedCrop (#151) 2021-01-25 20:28:16 +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
LXXXXR 13c1210741
[Feature] Add thrs in eval_metrics (#146) 
* support thr

* replace thrs with thr

* fix docstring

* minor change

* revise according to comments

* revised according to comments

* revise according to comments

* rewrite basedataset.evaluate to avoid duplicate calculation

* minor change

* change thr to thrs

* add more unit test
 2021-01-25 17:54:22 +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
LXXXXR 8e990b5654
[Feature] Support support and class-wise evaluation results (#143) 
* support support, support class-wise evaluation results and move eval_metrics.py

* Fix docstring

* change average to be non-optional

* revise according to comments

* add more unittest
 2021-01-19 16:42:16 +08:00
LXXXXR 4f8fc9cbf3
[Feature] Add analyze_results.py to select imgs for success/fail (#142) 
* add analyze_results.py to select imgs for success/fail

* minor change

* save imgs
 2021-01-19 15:49:14 +08:00
LXXXXR b98b317012
fix bug in vgg weight_init (#140) 2021-01-15 10:23:11 +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 63f38988eb
[Fix] Fix optional issues in docstring (#138) 
* fix optional issue in docstring

* revised according to comments

* add optional
 2021-01-14 11:09:08 +08:00
LXXXXR 6916f33d56
[Feature] Add asymmetric loss for multilabel task (#132) 
* add asymmetric loss

* minor change

* fix docstring

* do not apply sum over classes and fix docstring

* fix docstring

* fix weight shape

* fix weight shape

* add reference

* fix linkting issue

Co-authored-by: Y. Xiong <xiongyuxy@gmail.com>
 2021-01-11 11:22:22 +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 c07e60dd7d
[Feature] Support PASCAL VOC 2007 dataset for multilabel task (#134) 
* support voc

* minor change
 2021-01-11 11:08:59 +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
LXXXXR 9578bfa0f1
[Feature] Add focal loss for multilabel task (#131) 
* add focal loss

* apply class wise sum

* fix doctring

* do not apply sum over classes and fix docstring

* fix docstring

* fix weight shape

* fix weight shape
 2021-01-08 20:44:23 +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 1df10beaa1
Add evaluation metrics for multilabel task (#123) 
* add mean_ap

* add difficult_examples in mAP to support dataset without difficult_examples

* fix docstring

* add CP,CR,CF1,OP,OR,OF1 as multilabel metrics

* fix docstring

* temporary solution to ci until new version of mmcv is avaliable (#127)

* temporary solution to ci until new version of mmcv is avaliable

* temporary solution to ci until new version of mmcv is avaliable

* add mean_ap

* add difficult_examples in mAP to support dataset without difficult_examples

* fix docstring

* add CP,CR,CF1,OP,OR,OF1 as multilabel metrics

* fix docstring

* Swap -1 and 0 for labels

* Revised according to comments

* Revised according to comments

* Revised according to comments

* Revert "Revised according to comments"
It is suggested that we should not include paper from arxiv.
This reverts commit 48a781cd6a.

* Revert "Revert "Revised according to comments""

This reverts commit 6d3b0f1a7b.

* Revert "Revised according to comments"
It is suggested we should not cite paper from arxiv.
This reverts commit 120ecda884.

* Revised according to comments

* revised according to comments

* Revised according to comments
 2021-01-04 12:25:33 +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 996d2a7641
Revert "temporary solution to ci until new version of mmcv is avaliable (#127)" (#128) 
This reverts commit abb3234689.
 2020-12-30 17:11:33 +08:00
LXXXXR d291f2d889
Refactor docs (#126) 
* refactor docs

* revised according to comments
 2020-12-28 19:47:06 +08:00
LXXXXR abb3234689
temporary solution to ci until new version of mmcv is avaliable (#127) 
* temporary solution to ci until new version of mmcv is avaliable

* temporary solution to ci until new version of mmcv is avaliable
 2020-12-28 11:29:29 +08:00
congee 3e20b5efa6
add modelzoo statistics (#119) 2020-12-25 17:17:43 +08:00
LXXXXR 4203b94643
fix bug in eval_metrics (#122) 2020-12-23 16:20:47 +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 6331eb3e2f
[Bug] Fix bug in collect_results (#114) 
* fix bug in collect results

* fix compatibility issue in build.yml

* fix compatibility issue in build.yml

* revert changes in ci

* add comments
 2020-12-18 17:00:47 +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
LXXXXR 9a0c75ea41
Use old resolver in pip (#115) 
* use old resolver in pip

* use old resolver in pip
 2020-12-18 16:00:28 +08:00
LXXXXR f355f15485
Fix bug in gpu_ids in distributed training (#107) 
* update gpu_ids in distributed training

* move linear scaling rule after getting correct gpu_ids

* Remove support for autoscale_lr
 2020-12-09 16:28:03 +08:00
LXXXXR b1e91f256b
fix bug in recall and precision (#112) 2020-12-09 16:27:42 +08:00
LXXXXR 92438da12a
fix bug (#108) 2020-12-04 11:41:49 +08:00