Automated fruit recognition using EfficientNet and MixNet

摘要

The classification of fruits offers many useful applications in daily life, such as automated harvesting or building up stocks for supermarkets. Studies have been proposed to classify fruits from input images, exploiting image processing and machine learning techniques. Though a lot of improvements have been achieved in recent years, many approaches still suffer prolonged training/testing time, or a considerably high number of false positives. For several applications, it is crucial to provide users with not only precise but also real-time recommendations. In this paper, we propose a practical solution to fruit recognition by exploiting two recently-developed classifiers that have demonstrated themselves to be both effective and efficient. We adopted EfficientNet and MixNet, two families of deep neural networks to build an expert system being able to accurately and swiftly identify fruits. Such a system can be deployed onto devices with limited computational resources to prompt exact and timely recommendations. The approach's performance has been validated on a real dataset consisting of 48,905 images for training and 16,421 images for testing. The experimental results showed that the application of EfficientNet and MixNet on the considered dataset substantially improves the overall prediction accuracy in comparison to a well-established baseline.