公开(公告)号：CN117292313A

公开(公告)日：2023-12-26

申请号：CN202311219251.4

申请日：2023-09-20

Applicant: 上海应用技术大学

Inventor： 肖立中 ,  胡凡

IPC: G06V20/52 ,  G06V10/44 ,  G06V10/80 ,  G06V10/764 ,  G06V10/82 ,  G06N3/0464 ,  G06N3/048 ,  G06N3/08

Abstract: 本发明涉及一种基于改进YOLOv7模型的小目标漂浮垃圾检测方法，其特征在于，包括以下步骤：获取水面漂浮垃圾检测数据集；构建基于YOLOv7的小目标漂浮垃圾检测基准模型，引入空洞卷积改进颈部网络中的特征融合部分，使用空洞空间金字塔池化模块ASPP替换空间金字塔池化模块，并嵌入上下文内容增强模块CAM，用不同膨胀率的空洞卷积获取不同感受野的上下文信息，并从上到下注入特征融合网络以丰富上下文信息；对基于改进YOLOv7模型的小目标漂浮垃圾检测模型进行训练，将训练后的模型在测试集上进行检测并验证。解决了在复杂场景下检测特征不清晰的小型垃圾目标时，难以解决小尺寸目标特征信息丢失的问题。检测效果更好，漏检误检减少，检测物体的置信度也得到提高。