当前位置: 首页 > 文章 > 基于改进YOLOv7的复杂环境下红花采摘识别 农业工程学报 2023 (6) 169-176
Position: Home > Articles > 基于改进YOLOv7的复杂环境下红花采摘识别 Transactions of the Chinese Society of Agricultural Engineering 2023 (6) 169-176

基于改进YOLOv7的复杂环境下红花采摘识别

作  者:
王小荣;许燕;周建平;陈金荣
单  位:
新疆大学机械工程学院
关键词:
图像识别;图像处理;复杂环境;YOLOv7;注意力机制;多分类Focal Loss损失函数;红花采摘
摘  要:
针对光照、遮挡、密集以及样本数量不均衡等复杂环境造成红花机械化采摘识别不准问题,该研究提出一种基于YOLOv7的改进模型,制作红花样本数据集建立真实采摘的复杂环境数据,增加Swin Transformer注意力机制提高模型对各分类样本的检测精准率,改进Focal Loss损失函数提升多分类任务下不均衡样本的识别率。经试验,改进后的模型各类别样本的检测平均准确率达到88.5%,与改进前相比提高了7个百分点,不均衡类别样本平均精度提高了15.9个百分点,与其他模型相比,检测平均准确率与检测速度均大幅提升。改进后的模型可以准确地实现对红花的检测,模型参数量小,识别速度快,适合在红花采摘机械上进行迁移部署,可为红花机械化实时采摘研究提供技术支持。

相似文章

计量
文章访问数: 17
HTML全文浏览量: 0
PDF下载量: 0

所属期刊

推荐期刊