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

17 Commits (851b4385748ca59efd494ecc17ed54a64c2279aa)

Author SHA1 Message Date
Ezra-Yu 4b2128fdbd
[Fix] Fix bug when using `ClassBalancedDataset` (#555) 
* fix dataset wrapper bug and add unit tests

* fix dataset wrapper bug and add unit tests

* update unit tests

* fix lint

* align interface to mmdet

* update unit tests
 2021-11-26 09:40:00 +08:00
Ezra-Yu 2ce5825ef1
[Feature] Support ImageNet21k dataset. (#461) 
* add imagnet21k

* Update unit test

* update imaenet21k

* use slots

* use slots

* rename Data_item to ImageInfo

* add unit tests

* Update unit tests

* rm some print

* update unit tests

* fix lint

* remove default value of pipeline
 2021-10-28 15:22:08 +08:00
Ma Zerun f9eb9b409b
[Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
Azad Baykara 5c4da16084
[Docs] Fix typo 'metirc' (#411) 2021-08-17 10:35:26 +08:00
Ma Zerun 15cd34bbef
[Fix] Use zero as default value of `thrs` in metrics. (#341) 
* Use zero as default value of `thrs` in metrics. And it accepcts a number
instead of float now.

* Fix unit test comment

* Don't pass thrs if no thrs.
 2021-07-18 16:57:21 +08:00
Ma Zerun d04ebc1eb5
[Docs] Add API Reference in the docs (#342) 
* Add API inference in the docs and fix readthedocs config.

* Replace some relative link in docs.

* Format docstring for reStructuredText syntax.

* Fix vit paper link

* Fix docstring of `show_results` function in `BaseClassifier`.
 2021-07-14 15:06:50 +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 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 c07e60dd7d
[Feature] Support PASCAL VOC 2007 dataset for multilabel task (#134) 
* support voc

* minor change
 2021-01-11 11:08:59 +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 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 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
yanglei fbc68bf266 Add classifiers, heads, necks and losses 2020-07-07 19:32:06 +08:00
yanglei ba789da5c8 Add commonly used datasets 2020-07-01 16:09:06 +08:00
chenkai a99e6eb10d init commit 2020-05-21 21:21:43 +08:00