mmclassification

Commit Graph

Author	SHA1	Message	Date
Ma Zerun	3e198e63e8	[Docs] Update install tutorials. (#854 ) * [Docs] Update install tutorials. * [Docs] Improve dataset docs * Add option to show the results in demo. * fix typo	2022-06-01 18:31:57 +08:00
Hu Di	b4eefe4794	[Enhance] Support training on IPU and add fine-tuning configs of ViT. (#723 ) * implement training and evaluation on IPU * fp16 SOTA * Tput reaches 5600 * 123 * add poptorch dataloder * change ipu_replicas to ipu-replicas * add noqa to config long line(website) * remove ipu dataloder test code * del one blank line in test_builder * refine the dataloder initialization * fix a typo * refine args for dataloder * remove an annoted line * process one more conflict * adjust code structure in mmcv.ipu * adjust ipu code structure in mmcv * IPUDataloader to IPUDataLoader * align with mmcv * adjust according to mmcv * mmcv code structre fixed Co-authored-by: hudi <dihu@graphcore.ai>	2022-04-29 22:22:19 +08:00
mzr1996	5a67bc8a1a	[Fix] Fix unexpected modification on `image_demo.py`	2022-04-01 16:50:23 +08:00
Ezra-Yu	922a31a68e	[Feature] ADD RepMLP Backbone and checkpoints. (#709 ) * add repmlp * refactor backbone code * add checkpoint, readme, unit tests * unchange demo * improve docstring. * update pic * refactor patch embed * refactor unit tests * fix lint * update tools * update tools * Update checkpoint path Co-authored-by: mzr1996 <mzr1996@163.com>	2022-03-30 20:40:25 +08:00
Ezra-Yu	131d8c6296	[Feature] Add CAM visualization tool (#577 ) * add cam-grad tool * refactor cam-grad tool * add docs * update docs * Update docs and support Transformer * remove pictures and use link * replace example img and finish EN docs * improve docs * improve code * Fix MobileNet V3 configs * Refactor to support more powerful feature extraction. * Add unit tests * Fix unit test * fix distortion of visualization exapmles in docs * fix distortion * fix distortion * fix distortion * merge master * merge fix conficts * Imporve the tool * Support use both attribute name and index to get layer * add default get_target-layers * add default get_target-layers * update docs * update docs * add additional printt info when not using target-layers * Imporve docs * Fix enumerate list. Co-authored-by: mzr1996 <mzr1996@163.com>	2021-12-23 18:53:40 +08:00
Ma Zerun	f9eb9b409b	[Docs] Add Copyright information. (#413 )	2021-08-17 19:52:42 +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

8 Commits (v0.23.2)