Commit Graph

551 Commits (df6edd7f5a53cd6fa3a5c979fddf542fa926c671)
 

Author SHA1 Message Date
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
Y. Xiong d369645e23
bump version to 0.6.0 (#62)
* fix lint

* fix version path type

* del short_version

* setup: del ops; add python3.8

* del useless requirements

* setup: remove extra_requirements

* setup: totally remove build_requests

* setup: change to valid email
2020-10-11 00:12:04 +08:00
Y. Xiong c00ee6a876
workflow: modify build, add deploy (#61)
* workflow: modify build, add deploy

* fix type: mmseg -> mmcls

* simplify build by delete cuda related settings

* re-sort requirements by pre-commit hook

* build: support more pytorch versions

* build: fix torchvision version bug

* del useless test imports

* merge upstream

* build: del docstring checking to pass coverage check
2020-10-10 22:15:50 +08:00
Lei Yang 24fd4fb627
Visualize results on image demo (#58)
* visualize results on image demo

* add matplotlib in requirements
2020-10-10 16:33:27 +08:00
Y. Xiong 52d8ea4621
Merge pull request #53 from daavoo/fix-configs
fix outdated configs
2020-10-01 23:05:56 +08:00
David de la Iglesia Castro b577594d3d fix outdated configs 2020-10-01 16:15:14 +02:00
Lei Yang edffbcc486
Add pytorch2onnx (#20)
* add pytorch2onnx

* add pytorch2onnx API
2020-09-30 20:13:36 +08: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