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