【YOLOv7热力图可视化】参考链接非常棒,此博客用于记录一些细节(如果遇到报错:ValueError: too many values to unpack (expected 2))

在尝试将YOLOv7与GradCAM结合进行热力图可视化时,遇到了ValueError。问题主要出现在代码修改的部分,如expected3,got2和expected2的错误。解决方案包括在IDetect的fuseforward函数而非Detect的forward函数中进行修改,以及恢复YOLO模型的原始输出设置以避免过多的值被解包。成功改进后的模型需关注输出层位置。
摘要由CSDN通过智能技术生成

参考链接

细节

1. 直接跟着链接改完,发现报错ValueError: not enough values to unpack (expected 3, got 2)

  • 原因:修改部分1添加的四条语句,不应该在Detect类中的forward函数中操作,而是应该在IDetect类中的fuseforward函数中操作
    在这里插入图片描述

2. 会有检测框,去掉的方法

在这里插入图片描述
在这里插入图片描述

3. 改进之后的模型,应该注意输出层的位置

在这里插入图片描述

4. 遇到报错 ValueError: too many values to unpack (expected 2)

在这里插入图片描述

解决方法:还原models/yolo.py里面的out值,因为为了制作热力图,对IDetectfuseforward函数里面最后面的out进行了更改。因此做以下操作即可:

在这里插入图片描述

评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

孟孟单单

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值