Camellia oleifera Fruit Detection in Natural Scene Based on YOLO v5s

被引：0
|
作者：
Song H. [1 ,2 ]
Wang Y. [1 ,2 ]
Wang Y. [1 ,2 ]
Lü S. [1 ,2 ]
Jiang M. [1 ,2 ]
机构：
[1] College of Mechanical and Electronic Engineering, Northwest A&F University, Shaanxi, Yangling
[2] Key Laboratory of Agricultural Internet of Things, Ministry of Agriculture and Rural Affairs, Shaanxi, Yangling
来源：
Nongye Jixie Xuebao/Transactions of the Chinese Society for Agricultural Machinery | 2022年 / 53卷 / 07期
关键词：
Camellia oleifera fruits; data enhancement; deep learning; natural scene; object detection; YOLO v5s;
D O I：
10.6041/j.issn.1000-1298.2022.07.024
中图分类号：
学科分类号：
摘要：
In view of the characteristics of small size, dense distribution and changeable color of Camellia oleifera fruit, in order to realize the rapid and accurate identification of Camellia oleifera fruit in complex natural scene, and determine the appropriate clamping position for the automatic oscillating harvesting device according to the density distribution of the fruit, the YOLO v5s convolutional neural network model was used to carry out research on the image detection method of Camellia oleifera fruit in the natural scene. Through data enhancement, totally 3 296 Camellia oleifera fruit images were obtained to make the PASCAL VOC data set. After 150 rounds of training, the optimal weight model was got. The accurate rate was 90. 73%, the recall rate was 98. 38%, the comprehensive evaluation index was 94. 4%, the average detection accuracy was 98. 71%, the single image detection time was 12. 7 ms, and the memory size of the model was 14. 08 MB. Compared with the current mainstream first-stage detection algorithms YOLO v4 - tiny and RetinaNet, its accuracy rate was increased by 1. 99 percentage points and 4. 50 percentage points, the recall rate was increased by 9. 41 percentage points and 10. 77 percentage points, and the time was reduced by 96. 39% and 96. 25%, respectively. In addition, the weight file of the YOLO v5s model was small, indicating that its network was simpler and had the advantage of rapid deployment. It could be transplanted to edge devices in the future to provide algorithm reference for the vision system of the Camellia oleifera fruit automatic harvesting device. Through comparative experiment, the results also showed that the model can achieve high-precision recognition and positioning of fruits in dense, occluded, dim environments and fuzzy blur conditions, and it had strong robustness. The research results can provide a reference for the research of mechanical harvesting of Camellia oleifera fruit under the natural complex environment. © 2022 Chinese Society of Agricultural Machinery. All rights reserved.
引用
收藏
页码：234 / 242
页数：8
相关论文
共 28 条
  • [1] WU Delin, ZHAO Enlong, JIANG Shan, Et al., Optimization analysis and test of canopy vibration parameters of camellia fruit based on double pendulum model [J], Transactions of the Chinese Society for Agricultural Machinery, 52, 12, pp. 96-104, (2021)
  • [2] WU Delin, LI Chao, CAO Chengmao, Et al., Analysis and experiment of the operation process of branch-shaking type camellia oleifera fruit picking device [J], Transactions of the CSAE, 36, 10, pp. 56-62, (2020)
  • [3] LI Han, TAO Hanhu, CUI Lihao, Et al., Recognition and localization method of tomato based on SOM - K - means algorithm [J], Transactions of the Chinese Society for Agricultural Machinery, 52, 1, pp. 23-29, (2021)
  • [4] YUAN Jin, Research progress analysis of robotics selective harvesting technologies [J], Transactions of the Chinese Society for Agricultural Machinery, 51, 9, pp. 1-17, (2020)
  • [5] WANG Wenjie, GONG Liang, YANG Zhiyu, Et al., Tomato fruit recognition based on multi-source fusion image segmentation algorithm in open environment, Transactions of the Chinese Society for Agricultural Machinery, 52, 9, pp. 156-164, (2021)
  • [6] YANG Changhui, LIU Yanping, WANG Yi, Et al., Research and experiment on recognition and location system for citrus picking robot in natural environment, Transactions of the Chinese Society for Agricultural Machinery, 50, 12, pp. 14-22, (2019)
  • [7] XIONG Juntao, HE Zhiliang, TANG Linyue, Et al., Visual localization of disturbed grape picking point in non-structural environment [J], Transactions of the Chinese Society for Agricultural Machinery, 48, 4, pp. 29-33, (2017)
  • [8] LIU Tianzhen, TENG Guifa, YUAN Yingchun, Et al., Winter jujube fruit recognition method based on improved YOLO v3 under natural scene, Transactions of the Chinese Society for Agricultural Machinery, 52, 5, pp. 17-25, (2021)
  • [9] LIU Xiaogang, FAN Cheng, LI Jianian, Et al., Identification method of strawberry based on convolutional neural network [J], Transactions of the Chinese Society for Agricultural Machinery, 51, 2, pp. 237-244, (2020)
  • [10] CHEN Bin, RAO Honghui, WANG Yulong, Et al., Study on detection of camellia fruit in natural environment based on Faster - RCNN [J], Acta Agriculturae Jiangxi, 33, 1, pp. 67-70, (2021)
123