摘要
本发明公开了一种提高小目标检测精度和坐标变化速度的方法,该方法包括下述步骤:获取苹果图像数据并进行数据增强;对数据增强后的苹果图像数据进行数据预处理,并划分数据集;构建苹果图像识别网络模型,将YOLOv5网络模型中SPP模块后的第一个Conv卷积模块更换为Coord‑Conv卷积模块,并在三个检测头前添加Coord‑Conv卷积模块,联合Shape‑IOU和IOU方法对网络模型进行优化;基于训练集对苹果图像识别网络模型进行训练;基于测试集对苹果图像识别网络模型进行测试,输出图像识别的准确率;基于训练后的苹果图像识别网络模型得到预测的图像识别结果。本发明提高了坐标变换的速度和小目标检测的精度。
技术关键词
卷积模块
图像
网络模型训练
坐标
退火算法
精度
数据获取模块
像素
速度
测试模块
模板
输出模块
优化器
检测头
训练集
标签
指数
系统为您推荐了相关专利信息
发光鼠标垫
发光组件
RGB芯片
灯片
柔性LED灯带
智能行车控制系统
扫描模块
三维模型信息
雷达
控制模块