mmclassification/mmcls/datasets
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
..
pipelines Add albumentations (#45) 2020-09-22 17:35:39 +08:00
samplers Add classifiers, heads, necks and losses 2020-07-07 19:32:06 +08:00
__init__.py Add commonly used datasets 2020-07-01 16:09:06 +08:00
base_dataset.py Add model inference (#16) 2020-09-30 19:00:20 +08:00
builder.py Add classifiers, heads, necks and losses 2020-07-07 19:32:06 +08:00
cifar.py Add model inference (#16) 2020-09-30 19:00:20 +08:00
dataset_wrappers.py Add commonly used datasets 2020-07-01 16:09:06 +08:00
imagenet.py Add model inference (#16) 2020-09-30 19:00:20 +08:00
mnist.py Add model inference (#16) 2020-09-30 19:00:20 +08:00
utils.py replace urlretrieve with urlopen (#13) 2020-07-27 22:20:03 +08:00