Design of fruit classification system based on deep learning

This paper mainly studies the classification and detection of fruit based on deep learning. In order to solve the problem of current industrial demand, this paper proposes a new method for classification and detection of fruit based on YOLO. In the detection algorithm, the target detection and classification tasks are treated as a regression problem. It is convenient to transplant to different devices in the complex environment of industrial environment. A multi-feature fusion and multiple pre-selection frame prediction strategies to solve a variety of fruit classification problems. The test results show that the system can achieve the accuracy, real-time and compatibility required by the industry.