https://github.com/mynul-islam-madhurjo/art-recognizer

An image classifier that can classify 10 different types of arts around the world
https://github.com/mynul-islam-madhurjo/art-recognizer

art art-classification classification computer-vision deep-learning machine-learning

Last synced: about 1 month ago
JSON representation

An image classifier that can classify 10 different types of arts around the world

• Host: GitHub
• URL: https://github.com/mynul-islam-madhurjo/art-recognizer
• Owner: mynul-islam-madhurjo
• License: mit
• Created: 2023-12-10T17:01:17.000Z (about 1 year ago)
• Default Branch: main
• Last Pushed: 2023-12-23T13:32:53.000Z (about 1 year ago)
• Last Synced: 2023-12-24T09:30:06.140Z (about 1 year ago)
• Topics: art, art-classification, classification, computer-vision, deep-learning, machine-learning
• Language: Jupyter Notebook
• Homepage: https://mynul-islam-madhurjo.github.io/Art-Recognizer/
• Size: 40.7 MB
• Stars: 0
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE.txt

Awesome Lists containing this project

README

        
# Art-Recognizer

An image classification model from data collection, cleaning, model training, deployment and API integration. 


The model can classify 10 different types of arts 


The types are following: 


1. Cubism

2. Impressionism

3. Surrealism

4. Abstract Expressionism

5. Realism

6. Pop Art

7. Minimalism

8. Contemporary

9. Renaissance

10. Baroque

# Dataset Preparation

**Data Collection:** Downloaded from DuckDuckGo using term name 


**DataLoader:** Used fastai DataBlock API to set up the DataLoader. 


**Data Augmentation:** fastai provides default data augmentation which operates in GPU. 


Details can be found in `notebooks/Art_Recognizer.ipynb`

# Training and Data Cleaning

**Training:** Fine-tuned a resnet34 model for 5 epochs and got upto ~94% accuracy. 


**Data Cleaning:** This part took the highest time. Since I collected data from browser, there were many noises. Also, there were images that contained. I cleaned and updated data using fastai ImageClassifierCleaner. I cleaned the data each time after training or finetuning, except for the last time which was the final iteration of the model. 


# Model Deployment

I deployed to model to HuggingFace Spaces Gradio App. The implementation can be found in `deployment` folder or [here](https://huggingface.co/spaces/mynul-islam-madhurjo/arts-recognizer). 




# API integration with GitHub Pages

The deployed model API is integrated [here](https://mynul-islam-madhurjo.github.io/Art-Recognizer/) in GitHub Pages Website. Implementation and other details can be found in `docs` folder.