mmclassification

Commit Graph

Author	SHA1	Message	Date
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	c0e7512969	support options to override configs in inference.py (#96 )	2020-11-25 16:14:01 +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
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	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
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
yanglei	fbc68bf266	Add classifiers, heads, necks and losses	2020-07-07 19:32:06 +08:00
chenkai	70cda9c75e	fix linting	2020-05-21 21:27:54 +08:00
chenkai	a99e6eb10d	init commit	2020-05-21 21:21:43 +08:00

10 Commits (04cdbe4d66d5f60b88ffee84a949d6fc6df12a79)