121 Commits

Author SHA1 Message Date
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
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
b98b317012
fix bug in vgg weight_init (#140) 2021-01-15 10:23:11 +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
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
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 48a781cd6aca7be40821b55679bebb30295d4336.

* Revert "Revert "Revised according to comments""

This reverts commit 6d3b0f1a7b52afc8ed7a01a28e2911afa4f67c2b.

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

* 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
4203b94643
fix bug in eval_metrics (#122) 2020-12-23 16:20:47 +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
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
Ü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
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
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
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
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
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
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
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
anthracene
fef5202257
Fix crash in CenterCrop transform when image is greyscale (#40) 2020-08-31 00:04:39 +08:00
Xiaojie Li
7af9419ffa
Fix init_weights in 'shufflenet_v2.py'. (#29)
* fix init_weights in shufflenetv2

* fix doc

* fix doc

Co-authored-by: lixiaojie <lixiaojie@sensetime.com>
2020-08-13 09:49:41 +08:00
Lei Yang
408d92bcbe
revise docstr of alexnet to correct input_size (#26) 2020-08-12 16:14:19 +08:00
Lei Yang
c905decddf
Resize according to short edge (#22)
* resize according to short edge

* revise doc for resize

* fix unitest for resize

* resize short_edge when second value of size is -1

* rename short_edge to short_side
2020-08-11 16:04:45 +08:00
yl-1993
a0c814fecd add doc for backend 2020-08-05 15:35:40 +08:00
yl-1993
34efbc8a9d add pil backend for resize 2020-08-03 12:52:35 +08:00
Lei Yang
557a364d25
replace urlretrieve with urlopen (#13) 2020-07-27 22:20:03 +08:00
Kai Chen
0fb9b2f140
Update readme (#3)
* add badges and license

* minor format
2020-07-13 12:55:34 +08:00
yanglei
9a661ef981 Add ResNet_CIFAR 2020-07-12 00:06:56 +08:00
yanglei
e29882c8af fix old register 2020-07-12 00:06:56 +08:00
lixiaojie
c1d0090700 Change the init_weight for shufflenet models 2020-07-12 00:06:56 +08:00
yanglei
6968ad5b3b Add RandomFlip 2020-07-12 00:06:56 +08:00
yanglei
45812e87bd add weighted_loss 2020-07-12 00:06:55 +08:00
lixiaojie
1d5f3d8c24 update shufflenet_v1 2020-07-12 00:06:55 +08:00
lixiaojie
f6260b33bf Add RandomCrop, RandomResizedCrop, RandomGrayscale, impad 2020-07-12 00:06:55 +08:00
yanglei
8995e16834 Add AlexNet and LeNet5 2020-07-12 00:06:54 +08:00
yanglei
fbc68bf266 Add classifiers, heads, necks and losses 2020-07-07 19:32:06 +08:00
yanglei
910d13330a Add Resize and CenterCrop 2020-07-07 17:07:58 +08:00
yanglei
17092d8be4 Fix the mid_channels of SEResNeXt 2020-07-07 11:25:48 +08:00