【解决报错】YOLOv5 Grad CAM报错:keyerror:cv3

问题描述

YOLOv5添加注意力机制后 Grad CAM报错:keyerror:cv3
【解决报错】YOLOv5 Grad CAM报错:keyerror:cv3_第1张图片

原因分析:

增加注意力机制后,模型层数变化,打印出来,看一下在哪一层进行更改。
原始网络层:
【解决报错】YOLOv5 Grad CAM报错:keyerror:cv3_第2张图片
原始网络层:添加注意力机制后,网络层改变

【解决报错】YOLOv5 Grad CAM报错:keyerror:cv3_第3张图片

解决方案:

default='model_23_cv3_act'
改为:
default='model_24_cv3_act',

你可能感兴趣的:(深度学习,目标检测,YOLO)