Multi-class Image Classification Using Deep Learning Algorithm

Abstract

Classifying images is a complex problem in the field of computer vision. The deep learning algorithm is a computerized model simulates the human brain functions and operations. Training the deep learning model is a costly process in machine resources and time. Investigating the performance of the deep learning algorithm is mostly needed. The convolutional neural network (CNN) is most commonly used to build a structure of the deep learning models. In this paper convolutional neural network (CNN) model pre-trained on Image-Net is used for classification of images of the PASCAL VOC 2007 data-set. The transfer learning approach is used to improve the performance of the deep learning CNN model where classification works fairly well with the smallest amount of computation time and fewer machine resources. The behavior of the Deep learning CNN model is studied and the performance has been measured. The obtained results are compared with the obtained test results from the Super-vector coding of local image descriptors method, SVM method, and Region Ranking SVM method, which tested with the PASCAL VOC 2007 data-set. The final results evaluate the deep learning algorithm as a state-of-the-art method for an image classification task.