赵明岩,吴顺海,李一欣,等. 基于改进YOLOv5s的黑皮鸡枞菌检测方法[J]. 农业工程学报,2023,39(12):265-274. DOI: 10.11975/j.issn.1002-6819.202304104
    引用本文: 赵明岩,吴顺海,李一欣,等. 基于改进YOLOv5s的黑皮鸡枞菌检测方法[J]. 农业工程学报,2023,39(12):265-274. DOI: 10.11975/j.issn.1002-6819.202304104
    ZHAO Mingyan, WU Shunhai, LI Yixin, et al. Improved YOLOv5s-based detection method for termitomyces albuminosus[J]. Transactions of the Chinese Society of Agricultural Engineering (Transactions of the CSAE), 2023, 39(12): 265-274. DOI: 10.11975/j.issn.1002-6819.202304104
    Citation: ZHAO Mingyan, WU Shunhai, LI Yixin, et al. Improved YOLOv5s-based detection method for termitomyces albuminosus[J]. Transactions of the Chinese Society of Agricultural Engineering (Transactions of the CSAE), 2023, 39(12): 265-274. DOI: 10.11975/j.issn.1002-6819.202304104

    基于改进YOLOv5s的黑皮鸡枞菌检测方法

    Improved YOLOv5s-based detection method for termitomyces albuminosus

    • 摘要: 为解决黑皮鸡枞菌种植环境下背景土壤与菌菇辨识度较低、样本分布密集、类间相互遮挡等问题,该研究提出一种基于改进YOLOv5s的目标检测方法。首先,在骨干网络中融入RFBSE模块,使网络关注重点区域,通过施加通道注意力机制,增强对黑皮鸡枞菌特征表达能力;其次,设计多分支采样DCSPP池化模块,加强局部信息与全局信息的融合;第三,在颈部网络采用RFP结构,通过添加额外反馈信息促进语义信息传递,增强鸡枞菌样本密集遮挡场景下的检测能力,对RFP结构级联方式及网络间融合结构做轻量化处理,降低参数计算量和内存使用。试验结果表明,通过添加RFBSE模块,多分支池化模块以及采用递归金字塔结构对模型检测能力均有不同提升效果,最终模型平均精度均值mAP、精确率、召回率分别达到90.8%、86.5%、84.8%。对比原YOLOv5s模型算法,mAP、精确率、召回率分别提高2.7、3.8、3.9个百分点,并通过生成热力图提高模型检测过程的可解释性。试验结果表明改进后的模型可在复杂环境下准确、快速地识别黑皮鸡枞菌,为黑皮鸡枞菌采摘机器人的开发提供技术支持。

       

      Abstract: Termitomyces albuminosus is one species of agaric fungus in the family Agaricaceae in food production. However, it is a high demand to increase the current model detection accuracy, particularly under the complex planting environment, such as the variety of light and dark, the low recognition of soil and termitomyces albuminosus, the dense growth distribution and the serious shelter. In this study, target detection was proposed using an improved YOLOv5s. Firstly, RFBSE (Receptive Field Block Squeeze and Excitation) module was integrated into the backbone network. The human sensory field was then simulated to enhance the consistent contribution of different pixels to neural nodes in the process of network feature extraction. The edge features were highlighted to focus on the key areas. The irrelevant disturbances were suppressed using the module, such as the background. The attention mechanism of the channel was applied to gain the weight of different channels for the adaptive calibration of channel characteristic response. The channel was also enhanced to contain the important characteristic information of termitomyces albuminosus. As such, the high characteristic expression was achieved in the termitomyces albuminosus. Secondly, a multi-branch sampling DCSPP (Double Conv Spatial Pyramid Pooling) Pooling module was designed to perform the multiple sampling, in order to fuse the multiple receptive fields and then strengthen the relation between local and global information. The expression ability of the feature layer was enriched to improve the detection accuracy. Thirdly, the RFP (Recursive Feature Pyramid) structure was adopted in the neck network. The number of samples cannot increase, due to the usual interclass occlusion between termitomyces albuminosus. Previously, the network paid attention to the same image twice, because the context information around the occlusion samples was very important, and the feedback feature layer generated in the FPN structure was re-fed back to the backbone network for the Recursive computation. The neuronal activation was able to learn the correspondence and selectively inhibit, in order to improve the detection ability of the dense occluded sample of termitomyces albuminosus. The semantic information transmission was also promoted to enhance the context information near the occluded sample. At the same time, the cascaded RFP structure and the network fusion structure were lightened to reduce the calculation of parameters and memory usage. The ablation results showed that the RFBSE module, multi-branch pool module, and recursive pyramid structure shared different effects on the model. Specifically, the average precision mAP, precision, and recall rate of the final model reached 90.8%, 86.5%, and 84.8%, respectively. Each index of the improved model was improved, compared with the original one. The higher quality of the bounding box was achieved to detect the occluded target, where the mAP, accuracy, and recall were improved by 2.7, 3.8, and 3.9 percentage points, respectively. At last, the detection test of termitomyces albuminosus was carried out in different environments and shelter conditions by hardware platform model deployment. The visualized results showed that the detection rate of the model was more than 90%, which verified the validity of the model. The experimental results show that the improved model can be expected to accurately and rapidly identify the termitomyces albuminosus in a complex environment. The finding can provide technical support for the development of the termitomyces albuminosus harvesting robots.

       

    /

    返回文章
    返回