3 Class Image Classification CNN Model

Published: 28 February 2024| Version 1 | DOI: 10.17632/vcnv6y9r6z.1
taraneh saniei


This CNN model has been written and tested on Google Collab. It can be used for classification of compositional patterns containing color. It has been trained and tested on datasets containing paintings, graphic patterns, photography, interior architecture and building facades providing reliable outcomes. The model employs a ResNet network and can be run in any system as it needs the least space for being loaded and run. It also functions quite fast. The author suggests to use more than 10 epochs when training for better results. The model also can be merged with a similar structured MobileNet model.



Photography, Visual Analytics, Image Classification, Building Facade, Graphic Design, Convolutional Neural Network, Deep Learning, Interior Design, Automated Machine Learning