公开(公告)号：CN116883992A

公开(公告)日：2023-10-13

申请号：CN202310918241.3

申请日：2023-07-25

Applicant: 安徽农业大学

Inventor： 李科 ,  刘兵凯 ,  吴洪伟 ,  唐克青 ,  代遥 ,  徐启涛

IPC: G06V20/68 ,  G06V20/40 ,  G06V10/77 ,  G06V10/764 ,  G06V10/82 ,  G06N3/0464 ,  G06N3/048 ,  G06N3/045 ,  G06N3/08

Abstract: 本发明公开了一种基于改进YOLOv5x的草莓果实检测方法，包括：草莓图像预处理：草莓图像采集；样本数据增强，对提取的多张图片进行缩放、裁切和色调调整，再通过Mosaic混合方法来处理数据集；数据标注，使用开源图像标注工具Labelimg进行标注，将图片中的草莓对象位置和草莓分类信息标记出来；根据SD‑YOLOv5x的网络模型对草莓数据集进行训练。有益效果是：基于现有YOLOv5x模型的主干网络，引入了特征提取CCH模块和NAM注意力模块，结果表明经过改进的SD‑YOLOv5x网络在mAP、Precision和Recall方面都有很好的表现，与现有YOLOv5x模型相比有很大的提高。可以观察到改进后的SD‑YOLOv5x模型在目标检测框的精准性、置信度和检测效果等方面均表现得更加出色。