mmrazor/tools
whcao 1e8f886523
[Feature]Feature map visualization (#293)
* WIP: vis

* WIP: add visualization

* WIP: add visualization hook

* WIP: support razor visualizer

* WIP

* WIP: wrap draw_featmap

* support feature map visualization

* add a demo image for visualization

* fix typos

* change eps to 1e-6

* add pytest for visualization

* fix vis hook

* fix arguments' name

* fix img path

* support draw inference results

* add visualization doc

* fix figure url

* move files

Co-authored-by: weihan cao <HIT-cwh>
2022-10-26 13:26:20 +08:00
..
misc [fix] fix mmcv mmengine (#242) 2022-08-24 09:58:11 +08:00
model_converters [CI] Add circle ci (#257) 2022-08-30 20:20:10 +08:00
visualizations [Feature]Feature map visualization (#293) 2022-10-26 13:26:20 +08:00
dist_test.sh
dist_train.sh
get_channel_units.py merge pruning into dev-1.x (#312) 2022-10-10 17:30:25 +08:00
slurm_test.sh
slurm_train.sh
test.py [Fix] Fix mmcls import error (#206) 2022-07-26 15:33:11 +08:00
train.py [Feature] Support unroll with MMDDP in darts algorithm (#210) 2022-10-14 17:41:11 +08:00