yizt / Grad-CAM.pytorch

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

Repository from Github https://github.comyizt/Grad-CAM.pytorchRepository from Github https://github.comyizt/Grad-CAM.pytorch

您好,在运行demo_retinanet.py时

yxheartipp opened this issue · comments

您好,在运行demo_retinanet.py时

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