公开(公告)号：CN117710965A

公开(公告)日：2024-03-15

申请号：CN202410028101.3

申请日：2024-01-09

Applicant: 福建农林大学

Inventor： 叶大鹏 ,  景均 ,  谢立敏

IPC: G06V20/68 ,  G06V10/42 ,  G06V10/44 ,  G06V10/762 ,  G06V10/80 ,  G06V10/82 ,  G06N3/0464 ,  G06N3/08

Abstract: 本发明提出基于改进YOLOv5的小目标检测方法，包括以下步骤；步骤S1：输入阶段：聚类出适用的预测框尺寸；步骤S2：特征提取阶段：建立改进YOLOv5s的主干网络，优化模型局部、全局特征提取能力，提高精度和运算速度；步骤S3：特征聚合阶段：添加注意力机制NAM，剔除冗余特征信息，加强关键特征的利用；步骤S4：损失函数：以WIoU损失函数代替原有CIoU损失函数，关注边界框的特征学习，同时提高模型收敛速度和鲁棒性；步骤S5：检测阶段：扩大感受野提高检测精度；在尽可能多保留正确预测框的同时消除冗余预测框；利用多尺度训练策略得到多个子模型，提高模型检测精度；本发明对检测任务中的小目标检测进行改进优化，可应用于多种环境下食用菌的自动识别采摘等领域。