Fasterrcnn+fpn 可以用吗

yizt / Grad-CAM.pytorch

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

Apache License 2.0

715 stars 166 forks source link

Open zhaoyanjoin opened 3 years ago

zhaoyanjoin commented 3 years ago

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

RuoyuChen10 commented 3 years ago

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

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