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