基于SCG-YOLOv5n的收获期澳洲坚果检测算法  

Detection algorithm of macadamia nut at havesting stage based on SCG-YOLOv5n

作　　者：张慧蒙 何超[1] 徐嘉雯 罗鑫 荣剑[1] 刘学渊[1] Zhang Huimeng;He Chao;Xu Jiawen;Luo Xin;Rong Jian;Liu Xueyuan(College of Mechanical Engineering and Transportation,Southwest Forestry University,Kunming,650224,China)

机构地区：[1]西南林业大学机械与交通学院,昆明市650224

出　　处：《中国农机化学报》2024年第4期214-221,共8页Journal of Chinese Agricultural Mechanization

基　　金：云南省教育厅科学研究基金项目(2023Y0758);云南省科技厅科技计划项目(202301BD070001-041)。

摘　　要：为实现自然环境下澳洲坚果的快速准确检测,针对收获期澳洲坚果果皮与枝叶颜色相似、体积小、病害果混杂难识别的问题,提出一种基于SCG-YOLOv5n的收获期澳洲坚果检测算法。该方法运用数据增强,提高模型鲁棒性;在YOLOv5n的骨干网络引入SimAM注意力机制,增强有效特征的提取能力;在FPN结构中引入CARAFE上采样,强化目标感知能力;使用GSConv轻量级卷积替换部分卷积层,减轻模型的参数量并实现高效特征融合,提高检测速度和检测精度。结果表明,改进后的SCG-YOLOv5n澳洲坚果检测算法对收获期的青皮澳洲坚果和病害澳洲坚果的检测平均精度AP分别为94.8%、97.9%,单张图像平均时间为5.33 ms,比YOLOv5n模型高出2.1%、1.3%,检测速度提升15.8%。该算法可以高效检测澳洲坚果,为后续自动化采摘提供技术参考。In order to achieve fast and accurate detection of macadamia nuts in natural environment,a macadamia detection algorithm based on SCG-YOLOv5n during the harvesting period is proposed,aiming at the problems of the similar color of macadamia nut peel and branch leaves during harvesting period,small size and difficult identification of mixed diseased fruit.The method uses data augmentation to improve model robustness,introduces SimAM attention mechanism in the backbone network of YOLOv5n to enhance the extraction of effective features,introduces CARAFE up sampling in the FPN structure to strengthen target perception,uses GSConv lightweight convolution to replace some convolutional layers to reduce the number of model parameters and achieve efficient feature fusion to improve detection speed and detection accuracy.The results show that the improved SCG-YOLOv5n macadamia detection algorithm has an average accuracy AP of 94.8%and 97.9%for the detection of green macadamia nuts and diseased macadamia nuts during the harvest period,respectively,and the average time of a single image is 5.33 ms,which is 2.1%and 1.3%higher than the YOLOv5n model,and the detection speed is improved by 15.8%.The algorithm can efficiently detect macadamia nuts and provide technical reference for subsequent automated harvesting.

关 键 词：深度学习 澳洲坚果检测 数据增强 注意力机制 卷积神经网络设计 

分 类 号：S776.01[农业科学—森林工程] S225.93[农业科学—林学]