首页|期刊导航|农机化研究|基于改进YOLOv7-Tiny的树叶遮挡环境下红心李识别

基于改进YOLOv7-Tiny的树叶遮挡环境下红心李识别OA

Red-heart Plum Recognition Based on Improved YOLOv7-Tiny Under Leaf Occlusion

中文摘要英文摘要

在果园红心李检测任务中,首要任务是准确识别红心李.然而由于红心李枝叶茂盛、果实重叠,增加了识别的难度.基于此,通过修改YOLOv7-Tiny模型的主干,来提升遮挡环境下果实检测的精度.首先,在MobileNetV3 主干的Mo-bileBottleneck(Bneck)模块中将SE注意力机制修改为NAM注意力机制,关注训练过程调整权重的信息,提高对果实关键特征的检测,构建新的NBneck模块;然后,在MobileNetV3 主干中加入Diverse Branch Block(DBB)模块,增强对被遮挡果实不明显特征的提取能力;最后,构建新的主干DN-MBV3 来替换YOLOv7-Tiny的原有主干网络,并减少网络的参数量.在相同试验条件下,与SSD、YOLOv4-Tiny、EfficientDet、Faster-RCNN等模型相比,改进的YOLOv7-Tiny精度明显优于其他网络.与YOLOv7-Tiny相比,红心李的均值平均精度(mAP)提高了 4.89 个百分点,召回率提升了 11.59 个百分点,同时模型体积减小了 4.3 MB,具有检测精度高、尺寸小等优点.

Identifying accurately red-hearted plums was the priority task in red-hearted plum detection.However,the luxuriant branches and overlapping fruits increased the difficulty of recognizing red-hearted plums.Based on this,the backbone of the YOLOv7-Tiny model was modified to improve the accuracy of fruit detection in occluded environments.First,the SE attention mechanism was modified to the NAM attention mechanism in the MobileBottleneck(Bneck)mo dule of the MobileNetV3 backbone,focusing on the information of adjusting the weights during the training process to im-prove the detection of critical features of the fruits and the new NBneck module was constructed.In addition,the Diverse Branch Block(DBB)module was added to the MobileNetV3 backbone to enhance the ability to extract features that were not obvious to the occluded fruits,and ultimately,a new backbone DN-MBV3 was constructed to replace the original backbone network of the YOLOv7-Tiny and to reduce the number of parameter in the network.Under the same experi-mental conditions,the improved YOLOv7-Tiny accuracy was better than that of other networks,including SSD,YOLOv4-Tiny,EfficientDet,Faster-RCNN,and other models.Compared with YOLOv7-Tiny,the mean average accura-cy(mAP)of the red-hearted plum was improved by 4.89 percentage points,the recall rate was increased by 11.59 percentage points,and the model volume was reduced by 4.3 MB,which had the advantages of high detection accuracy and small size.

Zhang Xiaobin;Zhao Pengfei;Chen Zhenlei;Han Jiangjie;Qian Mengbo

School of Opto-Mechanical Engineering,Zhejiang Agriculture and Forestry University,Hangzhou 311300,ChinaSchool of Opto-Mechanical Engineering,Zhejiang Agriculture and Forestry University,Hangzhou 311300,ChinaSchool of Opto-Mechanical Engineering,Zhejiang Agriculture and Forestry University,Hangzhou 311300,ChinaSchool of Opto-Mechanical Engineering,Zhejiang Agriculture and Forestry University,Hangzhou 311300,ChinaSchool of Opto-Mechanical Engineering,Zhejiang Agriculture and Forestry University,Hangzhou 311300,China

农业科技

红心李检测YOLOv7-Tiny遮挡目标检测

red-hearted plum detectionYOLOv7-Tinyocclusiontarget detection

《农机化研究》 2026 (4)

225-231,7

国家自然科学基金项目(51875531),浙江省"尖兵""领雁"研发攻关计划项目(2022C02057)

10.13427/j.issn.1003-188X.2026.04.027

评论