关键词:
海滩垃圾
环境污染
目标检测
QARepVGG
注意力机制
摘要:
海滩垃圾污染对海洋生态系统和人类健康构成巨大威胁,海滩垃圾监测和清理是一项繁重且复杂的任务。传统的人工调查存在监测效率低、检测范围小和时效性差等问题,为了解决这些问题,本文提出一种基于改进YOLOv7的海滩垃圾检测方法。首先,YOLOv7与量化感知RepVGG(QARepVGG)相结合,实现快速计算并降低模型参数量。其次,加入简单注意力机制(simple attention mechanism,SimAM),增强网络对图像感兴趣区域的特征提取能力。最后,结合双向特征金字塔网络(bi-directional feature pyramid network,BiFPN)结构,改进原有路径聚合网络(path aggregation network,PAN),提高网络学习垃圾特征的效率,增强对不同尺寸垃圾的识别能力。在自建数据集上的实验结果表明:(1)改进模型对8类海滩垃圾有良好的检测能力;(2)与YOLOv7相比,改进模型的总体平均精度均值(mean average precision,mAP)提升5.8%,每秒传输帧数(frames per second,FPS)提高17,改进模型对泡沫、塑料类和纸制品垃圾的识别精度最高;(3)与几种流行的检测模型相比,改进模型的识别精度和效率最高。实际场景中的检测结果表明,改进模型能满足海滩垃圾实时性检测需求。