Multiclass Classification using Enhanced MobileNet V2 Architecture

Abstract

In computer vision applications including object detection, face recognition, image classification, etc., the Convolutional Neural Network (CNN) predominates. MobileNet v2 is one such CNN architecture which can significantly cut down the parameters and cost of calculation while giving comparatively higher accuracy. In this paper, we propose the enhanced MobileNet architecture which shows an improvement in the accuracy. We have applied transfer learning techniques to fine-tune the MobileNet V2 model on the food dataset. The proposed model’s generalization capacity, it’s robustness and recognition accuracy shows a considerable amount of improvement with respect to the base architecture. The findings demonstrate that the enhanced MobileNet architecture’s accuracy for food images recognition and classification has increased to 97.16%. Key Words: Convolutional Neural Network (CNN), MobileNet v2, transfer learning, image recognition and classification