农业机械学报2024，Vol.55Issue(5)：244-253,10.DOI:10.6041/j.issn.1000-1298.2024.05.023
基于频域数据增强与轻量化YOLO v7模型的成熟期香梨目标检测方法
Mature Stage Pear Detection Method Based on Frequency Domain Data Augmentation and Lightweight YOLO v7 Model
摘要
Abstract
In the practice of modern agricultural production,the method of harvesting agricultural products is gradually shifting towards mechanization and intelligence.An increasing number of robots are being introduced into actual production and progressively replacing traditional manual labor.However,in natural environments,factors such as weather,l ighting,the similarity in color between fruits and their backgrounds,and mutual occlusion between fruits and branches significantly increased the difficulty of fruit target detection.To accurately detect pears in natural environments,a lightweight pear detection method M-YOLO v7-SCSN+F was designed based on the YOLO v7-S foundational model.This model introduced MobileNetv3 into the YOLO v7-S model as its backbone feature extraction network,thereby reducing the number of parameters in the network.It incorporated a coordinate attention(CA)mechanism in the model's feature fusion layer to enhance the network's feature representation capabilities.The loss function CIoU in YOLO v7-S was replaced with SIoU,which was used in conjunction with the normalized Wasserstein distance(NWD)mechanism for small target detection,further improving the detection accuracy for fragrant pears.Based on the Fourier transform(FT)data augmentation method,new image data was generated by analyzing the frequency domain information of images and reconstructing the amplitude components,thereby enhancing the model's generalization ability.Experimental results showed that the improved M-YOLO v7-SCSN+F model achieved mean average precision(mAP),precision,and recall rates of 97.23％,97.63％and 93.66％,respectively,on the validation set,with a detection speed of 69.39 f/s.The proposed detection model improved performance compared with Faster R-CNN,SSD,YOLO v3,YOLO v4,YOLO v5s,YOLO v7-S,YOLO v8n and RT-DETR-R50 models on the validation set,with mean average precision(mAP)enhancements of 14.50,26.58,3.88,2.40,1.58,0.16,0.07 and 0.86 percentage points,respectively.Furthermore,the improved M-YOLO v7-SCSN+F model reduced its parameter count by 16.47 MB and 13.30 MB,respectively,when compared with the advanced YOLO v8n and RT-DETR-R50 detection models.The detection model introduced demonstrated a high degree of effectiveness in target detection for mature pears,offering a reference for detecting small objects with backgrounds of similar color,and provided effective technical support for the automation of pear harvesting.关键词
香梨/目标检测/YOLO v7/数据增强/傅里叶变换/注意力机制Key words
pear/object detection/YOLO v7/data augmentation/Fourier transform/attention mechanism分类
农业科技引用本文复制引用
郑文轩,杨瑛..基于频域数据增强与轻量化YOLO v7模型的成熟期香梨目标检测方法[J].农业机械学报,2024,55(5):244-253,10.基金项目
国家自然科学基金项目(51466014) （51466014）
