采用改进YOLOv4算法的大豆单株豆荚数检测方法  

Detection method of soybean pod number per plant using improved YOLOv4 algorithm

作　　者：郭瑞 于翀宇 贺红 赵永健 于慧[2] 冯献忠[2] Guo Rui;Yu Chongyu;He Hong;Zhao Yongjian;Yu Hui;Feng Xianzhong(School of Mechanical,Electrical and Information Engineering,Shandong University,Weihai 264209,China;Northeast Institute of Geography and Agroecology,Chinese Academy of Sciences,Changchun 130102,China)

机构地区：[1]山东大学机电与信息工程学院,威海264209 [2]中国科学院东北地理与农业生态研究所,长春130102

出　　处：《农业工程学报》2021年第18期179-187,共9页Transactions of the Chinese Society of Agricultural Engineering

基　　金：国家重点研发计划主要经济作物分子设计育种(No.2016YFD0101900)。

摘　　要：大豆单株豆荚数检测是考种的重要环节,传统方法通过人工目测的方式获取豆荚类型和数量,该方法费时费力且误差较大。该研究利用大豆单株表型测量仪采集到的表型数据,通过融合K-means聚类算法与改进的注意力机制模块,对YOLOv4目标检测算法进行了改进,使用迁移学习预训练,获取最优模型对测试集进行预测。试验结果表明,该研究模型的平均准确率为80.55%,数据扩充后准确率达到了84.37%,比育种专家目测准确率提高了0.37个百分点,若不考虑5粒荚,该研究模型的平均准确率为95.92%,比YOLOv4模型提高了10.57个百分点,具有更强的检测性能。在简单背景的摆盘豆荚检测中,该研究模型预测的平均准确率达到了99.1%,比YOLOv4模型提高了1.81个百分点,研究结果表明该模型在不同场景下的大豆豆荚检测中具有较强的泛化能力,可为大豆人工智能育种提供参考。Measuring pod number per plant has been one of the most important parts of the pod selection in the soybean growth period.However,traditional manual measurement is costly,time-consuming,and error-prone.Alternatively,artificial intelligence can ever-increasing be used to determine the type and quantity of each pod,thereby accurately predicting soybean yield in modern agriculture.In this study,a soybean phenotyping instrument was employed to collect the phenotype video of soybean plants,and then to process the phenotype data using the YOLOv4 dynamic object detection.The size of the initial anchor box and the complexity of image objects were also considered during the data set training and testing.Specifically,the COCO data set was selected for the prior box in the YOLOv4 model.The size of the objects varied in each category.K-means clustering was selected to adjust the size of original prior box for a higher accuracy of pod recognition.The size and position of container were calculated to test the original anchor frame suitable for pod detection.Accordingly,a total of 9 initial anchor frames were obtained.The average and maximum pooling operations of the global channel information were adopted to obtain the new weights and re-weighed the new weights to generate the output of module,in order to accurately locate pods for the better characterization ability of detection model.The improved attention mechanism module was integrated into the last layer of the backbone network in the YOLOv4 object detection.Migration learning was also utilized to pre-train the neural network for the optimal detection model in the prediction of test set.The experiment was performed on the Pytorch framework under the GPU(Nvidia GeForce RTX 2080 Ti).The parallel computing framework of CUDA10.1 and CUDNN deep neural network acceleration library were used to train the original and the improved YOLOv4 on Linux operating system.Experiment results showed that the improved model greatly improved the accuracy of pod detection.The mean average prec

关 键 词：图像识别 算法 大豆 豆荚检测 YOLOv4 K-MEANS聚类 注意力机制 

分 类 号：TP391.4[自动化与计算机技术—计算机应用技术] S126[自动化与计算机技术—计算机科学与技术]