yizt / Grad-CAM.pytorch

pytorch实现Grad-CAM和Grad-CAM++,可以可视化任意分类网络的Class Activation Map (CAM)图,包括自定义的网络;同时也实现了目标检测faster r-cnn和retinanet两个网络的CAM图;欢迎试用、关注并反馈问题...

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Fasterrcnn+fpn 可以用吗

zhaoyanjoin opened this issue · comments

是不是把网络换成我的就可以用了。。

是不是把网络换成我的就可以用了。。

对于有FPN的网络,你需要自己指定last layer,可以选roi_heads.box_pooler