lyuwenyu / RT-DETR

[CVPR 2024] Official RT-DETR (RTDETR paddle pytorch), Real-Time DEtection TRansformer, DETRs Beat YOLOs on Real-time Object Detection. 🔥 🔥 🔥
Apache License 2.0
2.31k stars 259 forks source link

resnet50的热力图 #191

Closed todesti2 closed 8 months ago

todesti2 commented 8 months ago

请问一下,如何在RT-DETR的resnet50的最后一层,画出grad cam热力图呢

lyuwenyu commented 8 months ago

这个目前仓库不支持的 你得找个第三方库去可视化,,可以在这找到最后一层的特征 https://github.com/lyuwenyu/RT-DETR/blob/main/rtdetr_pytorch/src/nn/backbone/presnet.py#L215

todesti2 commented 8 months ago

这个目前仓库不支持的 你得找个第三方库去可视化,,可以在这找到最后一层的特征 https://github.com/lyuwenyu/RT-DETR/blob/main/rtdetr_pytorch/src/nn/backbone/presnet.py#L215

哎 好吧,谢谢您

Ryanshuai commented 8 months ago

可以看下这个: https://github.com/jacobgil/pytorch-grad-cam

todesti2 commented 8 months ago

可以看下这个: https://github.com/jacobgil/pytorch-grad-cam

谢谢您,可是我用的是paddle框架……好多人都说做不了