一种基于改进YOLOv8s的疏果前石榴果实检测方法

    公开(公告)号:CN116958962A

    公开(公告)日:2023-10-27

    申请号:CN202310686822.9

    申请日:2023-06-12

    Abstract: 本发明公开了一种基于改进YOLOv8s的疏果前石榴果实检测方法,该方法包括:采集多张石榴的原始图像并使用“makesense”进行标注处理,生成“.txt”文件;对原始图像进行扩充,分别增强和减弱亮度、色度、对比度和锐度,同时添加高斯噪声,得到构建的数据集;通过添加注意力机制CBAM和小目标检测层来构建改进的YOLOv8s网络;定义损失函数Loss;将构建的数据集随机按7:3的比例划分训练集和验证集;使用训练集来训练改进的YOLOv8检测模型,同时开启在线数据增强和标签平滑并保存权重文件,通过验证集验证改进的YOLOv8s检测模型效果。采用该方法对数据集检测精度达到了99.1%,更好的提取了目标特征的同时,提升了目标检测的精度。

Patent Agency Ranking