open-mmlab / mmdetection

OpenMMLab Detection Toolbox and Benchmark
https://mmdetection.readthedocs.io
Apache License 2.0
29.21k stars 9.4k forks source link

How to integrate the CAM/GRAD-CAM and some heatmap visualization with the DETR , eg."DINO" ? I have seen so many ideas about CNN models but transformer_detection model #11779

Open xyb1314 opened 3 months ago

xyb1314 commented 3 months ago

thanks ! how to ?

xyb1314 commented 3 months ago

Can I use the mmpretrain/tools/visualization/vis_cam.py in the mmdetection package for CAM ?

xiuqhou commented 1 month ago

Hi @xyb1314 , we successfully integrated CAM and feature visualization into our DETR project. Maybe you can refer to it.

https://github.com/xiuqhou/Salience-DETR/blob/main/grad_cam/