search

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

您好，在运行demo_retinanet.py时 #31

Open yxheartipp opened 3 years ago

yxheartipp commented 3 years ago

出现torch.nn.modules.module.ModuleAttributeError: 'RetinaNet' object has no attribute 'in_features'问题 请问您应该怎么解决，感谢您。