Abstract:Contemporary Artificial Intelligence technologies allow for the employment of Computer Vision to discern good crops from bad, providing a step in the pipeline of selecting healthy fruit from undesirable fruit, such as those which are mouldy or damaged. State-of-the-art works in the field report high accuracy results on small datasets (<1000 images), which are not representative of the population regarding real-world usage. The goals of this study are to further enable real-world usage by improving generalisation with data augmentation as well as to reduce overfitting and energy usage through model pruning. In this work, we suggest a machine learning pipeline that combines the ideas of fine-tuning, transfer learning, and generative model-based training data augmentation towards improving fruit quality image classification. A linear network topology search is performed to tune a VGG16 lemon quality classification model using a publicly-available dataset of 2690 images. We find that appending a 4096 neuron fully connected layer to the convolutional layers leads to an image classification accuracy of 83.77%. We then train a Conditional Generative Adversarial Network on the training data for 2000 epochs, and it learns to generate relatively realistic images. Grad-CAM analysis of the model trained on real photographs shows that the synthetic images can exhibit classifiable characteristics such as shape, mould, and gangrene. A higher image classification accuracy of 88.75% is then attained by augmenting the training with synthetic images, arguing that Conditional Generative Adversarial Networks have the ability to produce new data to alleviate issues of data scarcity. Finally, model pruning is performed via polynomial decay, where we find that the Conditional GAN-augmented classification network can retain 81.16% classification accuracy when compressed to 50% of its original size.

Image based fruit category classification by 13-layer deep convolutional neural network and data augmentation

Fruit Category Classification Via an Eight-Layer Convolutional Neural Network with Parametric Rectified Linear Unit and Dropout Technique

Fruit Classification Using Convolutional Neural Network Via Adjust Parameter and Data Enhancement.

Fruit Classification Based on Six Layer Convolutional Neural Network

Automated Fruit Classification Based on Deep Learning Utilizing Yolov8

Apple Ripeness Identification Using Deep Learning

The Fruit Classification Algorithm Based on the Multi-Optimization Convolutional Neural Network

Implementation of Fruits Recognition Classifier using Convolutional Neural Network Algorithm for Observation of Accuracies for Various Hidden Layers

Fruit Quality Assessment with Densely Connected Convolutional Neural Network

Fruit Classification Model Based on Improved Darknet53 Convolutional Neural Network

Fruit Classification Using Computer Vision and Feedforward Neural Network

Fruit Image Classification Based on MobileNetV2 with Transfer Learning Technique

A Hybrid Deep Learning-Based Fruit Classification Using Attention Model and Convolution Autoencoder

A study on fruit classification using convolutional neural network for image recognition technology

Lightweight and Optimized Multi-Label Fruit Image Classification: A Combined Approach of Knowledge Distillation and Image Enhancement

DeepFruits: A Fruit Detection System Using Deep Neural Networks

Fruit Image Classification Model Based on MobileNetV2 with Deep Transfer Learning Technique

Fruit Detection and Three-Stage Maturity Grading Using CNN

Fruit Classification by Wavelet-Entropy and Feedforward Neural Network Trained by Fitness-Scaled Chaotic ABC and Biogeography-Based Optimization.

Fruit quality and defect image classification with conditional GAN data augmentation

Classification and Grading of Harvested Mangoes Using Convolutional Neural Network