科研之友
微信
新浪微博
Facebook
分享链接
摘要
针对光照、遮挡、密集以及样本数量不均衡等复杂环境造成红花机械化采摘识别不准问题，该研究提出一种基于YOLOv7的改进模型，制作红花样本数据集建立真实采摘的复杂环境数据，增加Swin Transformer注意力机制提高模型对各分类样本的检测精准率，改进Focal Loss损失函数提升多分类任务下不均衡样本的识别率。经试验，改进后的模型各类别样本的检测平均准确率达到88.5%，与改进前相比提高了7个百分点，不均衡类别样本平均精度提高了15.9个百分点，与其他模型相比，检测平均准确率与检测速度均大幅提升。改进后的模型可以准确地实现对红花的检测，模型参数量小，识别速度快，适合在红花采摘机械上进行迁移部署，可为红花机械化实时采摘研究提供技术支持。
- 单位
