A Study on CNN Transfer Learning for Image Classification

Many image classification models have been introduced to help tackle the foremost issue of recognition accuracy. Image classification is one of the core problems in Computer Vision field with a large variety of practical applications. Examples include: object recognition for robotic manipulation, pedestrian or obstacle detection for autonomous vehicles, among others. A lot of attention has been associated with Machine Learning, specifically neural networks such as the Convolutional Neural Network (CNN) winning image classification competitions. This work proposes the study and investigation of such a CNN architecture model (i.e. Inception-v3) to establish whether it would work best in terms of accuracy and efficiency with new image datasets via Transfer Learning. The retrained model is evaluated, and the results are compared to some state-of-the-art approaches.

[1]  Yuzuru Tanaka,et al.  Computational Collective Intelligence , 2016, Lecture Notes in Computer Science.

[2]  Diane Larsen-Freeman,et al.  Transfer of Learning Transformed. , 2013 .

[3]  Sunil Kumar Katiyar,et al.  Cellular neural network–based hybrid approach toward automatic image registration , 2013 .

[4]  Khalid M. Mosalam,et al.  Deep Transfer Learning for Image‐Based Structural Damage Recognition , 2018, Comput. Aided Civ. Infrastructure Eng..

[5]  Sergey Ioffe,et al.  Rethinking the Inception Architecture for Computer Vision , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[6]  Ngoc Thanh Nguyen,et al.  Transactions on Computational Collective Intelligence XXIII , 2016, Lecture Notes in Computer Science.

[7]  Dumitru Erhan,et al.  Going deeper with convolutions , 2014, 2015 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).