Using Channel and Network Layer Pruning Based on Deep Learning for Real-Time Detection of Ginger Images

Agriculture

Yuhua Li¹  Rui Xi¹  Yanqiang Wu¹  Xiaoyu Wang²  Jialin Hou²  Hongen Guo²  Lifa Fang²  Hua Zhang² 

[1] College of Mechanical & Electronic Engineering, Shandong Agricultural University, Tai’an 271018, China;Shandong Academy of Agricultural Machinery Sciences, Jinan 250100, China;

关键词: deep learning;    object detection;    network pruning;    ginger shoots;    ginger seeds;   

DOI  :  10.3390/agriculture11121190

来源: DOAJ

【 摘 要 】

Consistent ginger shoot orientation helps to ensure consistent ginger emergence and meet shading requirements. YOLO v3 is used to recognize ginger images in response to the current ginger seeder’s difficulty in meeting the above agronomic problems. However, it is not suitable for direct application on edge computing devices due to its high computational cost. To make the network more compact and to address the problems of low detection accuracy and long inference time, this study proposes an improved YOLO v3 model, in which some redundant channels and network layers are pruned to achieve real-time determination of ginger shoots and seeds. The test results showed that the pruned model reduced its model size by 87.2% and improved the detection speed by 85%. Meanwhile, its mean average precision (mAP) reached 98.0% for ginger shoots and seeds, only 0.1% lower than the model before pruning. Moreover, after deploying the model to the Jetson Nano, the test results showed that its mAP was 97.94%, the recognition accuracy could reach 96.7%, and detection speed could reach 20 frames·s⁻¹. The results showed that the proposed method was feasible for real-time and accurate detection of ginger images, providing a solid foundation for automatic and accurate ginger seeding.

【 授权许可】

Unknown