Automatic Fruits Freshness Classification Using CNN and Transfer Learning

Abstract

Fruit Freshness categorization is crucial in the agriculture industry. A system, which precisely assess the fruits’ freshness, is required to save labor costs related to tossing out rotten fruits during the manufacturing stage. A subset of modern machine learning techniques, which are known as Deep Convolution Neural Networks (DCNN), have been used to classify images with success. There have recently been many changed CNN designs that gradually added more layers to achieve better classification accuracy. This study proposes an efficient and accurate fruit freshness classification method. The proposed method has several interconnected steps. After the fruits data is gathered, data is preprocessed using color uniforming, image resizing, augmentation, and image labelling. Later, the AlexNet model is loaded in which we use eight layers, including five convolutional layers and three fully connected layers. Meanwhile, the transfer learning and fine tuning of the CNN is performed. In the final stage, the softmax classifier is used for final classification. Detailed simulations are performed on three publicly available datasets. Our proposed model achieved highly favorable results on all three datasets in which 98.2%, 99.8%, and 99.3%, accuracy is achieved on aforesaid datasets, respectively. In addition, our developed method is also computationally efficient and consumes 8 ms on average to yield the final classification result.