{"title":"Image Classification with Transfer Learning Using a Convolutional Neural Network","authors":"Merjem Bajramović, E. Žunić","doi":"10.1109/INFOTEH57020.2023.10094197","DOIUrl":null,"url":null,"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.","PeriodicalId":287923,"journal":{"name":"2023 22nd International Symposium INFOTEH-JAHORINA (INFOTEH)","volume":"34 1","pages":"0"},"PeriodicalIF":0.0000,"publicationDate":"2023-03-15","publicationTypes":"Journal Article","fieldsOfStudy":null,"isOpenAccess":false,"openAccessPdf":"","citationCount":"0","resultStr":null,"platform":"Semanticscholar","paperid":null,"PeriodicalName":"2023 22nd International Symposium INFOTEH-JAHORINA (INFOTEH)","FirstCategoryId":"1085","ListUrlMain":"https://doi.org/10.1109/INFOTEH57020.2023.10094197","RegionNum":0,"RegionCategory":null,"ArticlePicture":[],"TitleCN":null,"AbstractTextCN":null,"PMCID":null,"EPubDate":"","PubModel":"","JCR":"","JCRName":"","Score":null,"Total":0}
引用次数: 0
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.
京公网安备 11010802042870号
京ICP备2023020795号-1
分享
求助内容：
应助结果提醒方式：
扫码关注我们
