海洋环境下基于增强YOLOv7的垃圾目标检测
所属分类:技术论文
上传者:wwei
文档大小:3800 K
标签: EUIE eSE注意力 海洋垃圾检测
所需积分:0分积分不够怎么办?
文档介绍:针对海洋垃圾识别任务在实际应用中模型准确率不高的问题,提出一种基于优化YOLOv7的海洋垃圾识别算法。在图像增强部分,基于概率UIE的框架,通过添加eSE注意力减少特征信息的丢失。在损失函数部分,在IoU损失函数的基础上引入两层注意力机制的损失函数,将其与EIoU损失函数融合进一步提升模型的泛化能力。将该算法应用于海洋垃圾检测任务,并在基础数据集上对其进行评估。在YOLOTrashCan两个数据集上的平均精度均值指标分别达到69.5%、63.5%,相较于YOLOv7算法分别提升6%、1.6%。整体实验结果表明,所构建的算法能有效提升海洋垃圾检测的准确性。
现在下载
VIP会员,AET专家下载不扣分;重复下载不扣分,本人上传资源不扣分。