Open AccessCitrus recognition based on YOLOv4 neural network
Author(s) -
Changhong Wang,
Qiang Luo,
Xin Chen,
Bin Kong,
Haibao Wang
Publication year - 2021
Publication title -
journal of physics. conference series
Language(s) - English
Resource type - Journals
SCImago Journal Rank - 0.21
H-Index - 85
eISSN - 1742-6596
pISSN - 1742-6588
DOI - 10.1088/1742-6596/1820/1/012163
Subject(s) - computer science , robustness (evolution) , artificial intelligence , adaptability , artificial neural network , pattern recognition (psychology) , ecology , biochemistry , chemistry , biology , gene
In order to realize the accurate and rapid recognition of citrus fruit by autonomous picking robot in natural environment, a citrus recognition method based on YOLOv4 neural network is proposed. This algorithm improves the YOLOv4 model and uses the Kmeans++ algorithm to obtain the prior frame in the model to enhance the scale adaptability. Take citrus pictures independently and expand the data, use the LableImage platform to mark the data, and train the network model under the Darknet framework. According to the results, the citrus recognition model has good robustness and real-time performance to the common interference factors and their superposition in natural picking environment. The average recognition accuracy is 89.23 and the average detection speed is 60 ms. The average recognition accuracy is 89.23. It meets the requirements of real-time image recognition speed and accuracy of citrus picking robot.