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

问题描述

YOLOv5添加注意力机制后 Grad CAM报错:keyerror:cv3
在这里插入图片描述

原因分析:

增加注意力机制后,模型层数变化,打印出来,看一下在哪一层进行更改。
原始网络层:
在这里插入图片描述
原始网络层:添加注意力机制后,网络层改变

在这里插入图片描述

解决方案:

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

猜你喜欢

转载自blog.csdn.net/wahahaha116/article/details/129000560