Image Classification with Transfer Learning Using a Convolutional Neural Network

Abstract

Paper covers image classification using the Keras API in TensorFlow. The dataset used is a set of labelled images consisting of characters from the Pokémon media franchise. In order to artificially generate additional data, the process of data augmentation has been applied on the initial dataset to reduce overfitting. A comparison between DenseNet-121, DenseNet-169 and DenseNet-201 has been made to observe which of the models scores a greater accuracy. A Graphics Processing Unit (GPU) has been set up to work with TensorFlow in order to efficiently train the model.