mmpretrain

Commit Graph

Author	SHA1	Message	Date
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
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
Lei Yang	b2fc837e84	Rename configs with samples_per_batch x gpus (#37 ) * rename configs with samples_per_batch x gpus * revise docs accordingly	2020-08-23 23:53:51 +08:00
yanglei	563fa63d06	Add model_zoo docs	2020-07-12 00:06:57 +08:00
yanglei	d967b94439	Add getting_started docs	2020-07-12 00:06:53 +08:00

6 Commits (c00ee6a8763e3968512683bb8a8bf53351626450)