FPN-Based Small Orange Fruit Detection From Farm Images With Occlusion

Abstract

Fruit detection using deep learning is yielding very good performance, the goal of this work is to detect small fruits in images under these occlusion and overlapping conditions. The overlap among fruits and their occlusion can lead to false and missing detection, which decreases the accuracy and generalization ability of the model. Therefore, a small orange fruit recognition method based on improved Feature Pyramid Network was developed. To begin with, multi-scale feature fusion was used to fuse the detailed bottom features and high-level semantic features to detect small-sized orange to improve recognition rate. And then repulsion loss was used to take place of the original smooth L1 loss function. Besides, Soft non-maximum suppression was adopted to replace non-maximum suppression to screen the bounding boxes of orange to construct a recognition model of orange fruits. Finally, the network was trained and verified on the collected image data set. The results showed that compared with the traditional detection models, the mean average precision was improved from 79.7 to 82.8%.