A fruit detection algorithm based on R-FCN in natural scene

摘要

Aiming at the problem of poor precision and low efficiency of the vision system when agricultural robot picks fruit, this paper effectively fuses deep learning with machine vision, and proposes a new algorithm for fruit recognition and location, R-FCN, a deep learning algorithm combining regional suggestion network (RPN) and full convolutional neural network (FCN). The proposed algorithm uses FCN to convolve the input image to achieve feature extraction at the pixel level. Among them, the fusion of residual network can make the deep network have more abundant feature information for fruit recognition, and deconvolution can realize the visualization of detection results. Using RPN generates multiple candidate boxes on the feature map after convolution operation, this can effectively separate the image foreground and background. Training was conducted on the public COCO data set, and testing experiments were conducted on the different states of three different fruits. The experimental results show that the algorithm in this paper improves the detection accuracy by 0.71% and 0.33% respectively compared with the previous algorithm in apple and orange recognition. It can also have a recognition accuracy of 82.30% for banana, which is a large-scale fruit. For different input images, it can realize the visualization of fruit recognition and location, reduce the influence of branches and leaves occlusion, enhance the robustness of the system, and improve the efficiency of picking.