IMAGE CLASSIFICATION USING MODEL ENSEMBLING

ICTACT Journal on Image and Video Processing ( Volume: 12 , Issue: 4 )

Abstract

vioft2nntf2t|tblJournal|Abstract_paper|0xf4ff0e612c00000057c10e0001000100
Classifying images efficiently using various algorithms is very useful now-a-days given that the field of computer vision is growing rapidly. The research work highlighted in this paper focuses on the independent use of various models to classify images and then combining them together to form a better model in terms of performance than each of the individual models. The dataset used consists of 200 classes with 90,000 training images, 10,000 validation images and 10,000 test images. The data preparation step in this work involves resizing the images (data), shuffling them and transforming them into a data generator to provide input to the models. The images were also augmented using two different sets of image transformation effects to get more data for the models to train on. These data were then used to train five different models (one model trained from scratch and four other models using pre-trained weights and transfer learning) independently. The performance of each model was judged by checking two evaluation metrics –f1-score and categorical accuracy. The models were also tried to be fine-tuned to get a better performance, and finally the models were ensembled together to get a better categorical accuracy and f1-score on unseen (validation and test) data.

Authors

Debabrata Datta1, Anweshan Mukherjee2, Soumen Mukherjee3, Arup Kr. Bhattacharjee4, Anal Acharya5
St. Xavier's College, India1,2,5, RCC Institute of Information Technology, India3,4,

Keywords

Image Classification, Convolutional Neural Networks, Image Augmentation, Model Ensembling, F1-Score

Published By
ICTACT
Published In
ICTACT Journal on Image and Video Processing
( Volume: 12 , Issue: 4 )
Date of Publication
May 2022
Pages
2679-2692

ICT Academy is an initiative of the Government of India in collaboration with the state Governments and Industries. ICT Academy is a not-for-profit society, the first of its kind pioneer venture under the Public-Private-Partnership (PPP) model

Contact Us

ICT Academy
Module No E6 -03, 6th floor Block - E
IIT Madras Research Park
Kanagam Road, Taramani,
Chennai 600 113,
Tamil Nadu, India

For Journal Subscription: journalsales@ictacademy.in

For further Queries and Assistance, write to us at: ictacademy.journal@ictacademy.in