mmpretrain/tools/visualizations
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
..
vis_cam.py [Feature] Add CAM visualization tool (#577) 2021-12-23 18:53:40 +08:00
vis_lr.py [Feature] Add a tool to visualize learning rate in each iterations (#498) 2021-11-03 10:48:56 +08:00
vis_pipeline.py [Refactor] Use new API of matplotlib to handle blocking input in visualization. (#568) 2021-12-02 17:46:40 +08:00