A generic intelligent tomato classification system for practical applications using DenseNet-201 with transfer learning

• Main Authors: Tao Lu, Baokun Han, Lipin Chen, Fanqianhui Yu, Changhu Xue
• Format: Article
• Language: English
• Published: Nature Publishing Group 2021-08-01
• Series: Scientific Reports
• Online Access: https://doi.org/10.1038/s41598-021-95218-w

Abstract A generic intelligent tomato classification system based on DenseNet-201 with transfer learning was proposed and the augmented training sets obtained by data augmentation methods were employed to train the model. The trained model achieved high classification accuracy on the images of different quality, even those containing high levels of noise. Also, the trained model could accurately and efficiently identify and classify a single tomato image with only 29 ms, indicating that the proposed model has great potential value in real-world applications. The feature visualization of the trained models shows their understanding of tomato images, i.e., the learned common and high-level features. The strongest activations of the trained models show that the correct or incorrect target recognition areas by a model during the classification process will affect its final classification accuracy. Based on this, the results obtained in this study could provide guidance and new ideas to improve the development of intelligent agriculture.