Design of fruit classification system based on deep learning
计算机科学
人工智能
作者
Weiguo Zhang,Zhao Liu
标识
DOI:10.1117/12.2671086
摘要
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.