A novel method for vegetable and fruit classification based on using diffusion maps and machine learning

Abstract

Vegetable and fruit classification can help all links of agricultural product circulation to better carry out inventory management, logistics planning and supply chain coordination, and improve the efficiency and response speed of the supply chain. However, the current classification of vegetables and fruits mainly relies on manual classification, which inevitably introduces the influence of human subjective factors, resulting in errors and misjudgments in the classification of vegetables and fruits. In response to this serious problem, this research proposes an efficient and reproducible novel model to classify multiple vegetables and fruits using handcrafted features. In the proposed model, preprocessing operations such as Gaussian filtering, grayscale and binarization are performed on the pictures of vegetables and fruits to improve the quality of the pictures; statistical texture features representing vegetable and fruit categories, wavelet transform features and shape features are extracted from the preprocessed images; the feature dimension reduction method of diffusion maps is used to reduce the redundant information of the combined features composed of statistical texture features, wavelet transform features and shape features; five effective machine learning methods were used to classify the types of vegetables and fruits. In this research, the proposed method was rigorously verified experimentally and the results show that the SVM classifier achieves 96.25% classification accuracy of vegetables and fruits, which proves that the proposed method is helpful to improve the quality and management level of vegetables and fruits, and provide strong support for agricultural production and supply chain.

Keywords

• Vegetables and fruits classification
• Feature extraction
• Diffusion maps
• Machine learning