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

262 Commits (b30f79ea4ceb855d1a5540673d1951a3b981bf57)
 

Author SHA1 Message Date
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
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
Ülkü Tuncer Küçüktaş 6f7698cb7c
Update accuracy.py (#104) 
Co-authored-by: Ülkü Tuncer Küçüktaş <UlkuTuncerKucuktas@users.noreply.github.com>
 2020-12-01 14:25:18 +08:00
LXXXXR 5de2d2c0e9
Change to use mmcv.utils.collect_env to avoid duplicate codes (#103) 2020-12-01 14:24:08 +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
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
LXXXXR c0e7512969
support options to override configs in inference.py (#96) 2020-11-25 16:14:01 +08:00
LXXXXR 21fd5019fb
add macro-averaged precision,recall,f1 options in evaluation (#93) 
* add macro-averaged precision,recall,f1 options in evaluation

* remove unnecessary comments

* Revise according to comments

* Revise according to comments
 2020-11-25 16:13:54 +08:00
LXXXXR e75f2b7c35
add argument to override configs (#91) 2020-11-20 13:50:28 +08:00
LXXXXR 149ee3a30d
support img input as ndarray in inference.py (#87) 
* support img input as ndarray

* revise according to comments
 2020-11-19 18:58:25 +08:00
Wenwei Zhang fe45b241c5
Do not install mmcv through requirements (#90) 
* Delete the mmcv requirement to support mmcv-full installed environment

* install mmcv in CI

* fix mmcv version error

Co-authored-by: Xinran <lixinran@sensetime.com>
 2020-11-19 00:00:12 +08:00
LXXXXR 7636409b3b
add get_class in base_dataset (#85) 
* modify base_dataset

* revise according to the comments
 2020-11-12 14:22:02 +08:00
Lei Yang 909a6b9c3f
Add get_cat_ids in BaseDataset (#72) 
* visualize results on image demo

* add get_cat_ids in BaseDataset
 2020-10-26 14:04:10 +08:00
Lei Yang 784987fe9f
Fix slurm evaluation error during training (#69) 
* visualize results on image demo

* set round_up to True during test
 2020-10-20 15:04:39 +08:00
David de la Iglesia Castro df2189c6f6
Add missing CLASSES argument to dataset wrappers (#66) 
* Add missing classes in dataset wrappers

* Update tests
 2020-10-15 21:25:53 +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
Y. Xiong f7a916f309
fix pypi release error (#64) 2020-10-11 00:26:34 +08:00