基于改进yolov8网络的蔬菜田间杂草模型构建及检测方法
Abstract:
本发明公开一种基于改进yolov8网络的蔬菜田间杂草模型构建方法,包括:S1:采集蔬菜田间杂草图像,构建蔬菜杂草数据集,并按照预设的比例将蔬菜杂菜数据集划分为训练集、验证集和测试集;S2:构建改进的yolov8网络模型来建立蔬菜田间杂草检测模型;S3:将训练集输入至改进后的yolov8网络模型中进行训练,使用验证集来验证模型的表现并根据验证集的评估结果调整模型的参数,最后采用所述测试集来评估模型的性能,得到最终的蔬菜田间杂草检测模型;在所述步骤S2中包括了:S21:在原始yolov8网络模型中引入深度可分离卷积DSConv;S22:将neck层中的Concat模块替换为上下文交互融合模块CIFM;S23:将Head检测头替换为轻量化的LSCD检测头。本发明实现快速准确地检测出蔬菜与杂草。
Patent Agency Ranking
0/0