README.md

metadata

title: WikiArt Painting Classifier
emoji: 🎨
colorFrom: purple
colorTo: blue
sdk: gradio
sdk_version: 5.49.1
app_file: app.py
pinned: false
license: mit

WikiArt Painting Classifier - Interactive Demo

An interactive Gradio application for classifying paintings by artist, genre, and style.

Features

• Multi-Task Classification: Predicts artist, genre, and style simultaneously
• 129 Artists: Including Monet, Van Gogh, Picasso, Da Vinci, Rembrandt, and more
• 11 Genres: Portrait, landscape, abstract painting, still life, cityscape, etc.
• 27 Art Styles: Impressionism, Cubism, Renaissance, Baroque, Expressionism, and more
• Real-Time Predictions: Instant results with confidence scores

How to Use

1. Upload a painting image using the interface
2. Click "Classify Painting" or wait for automatic classification
3. View predictions for:
   • Top 10 likely artists with confidence scores
   • Genre classification with probabilities
   • Top 10 likely art styles with confidence scores

Model Details

This app uses the model hosted at: meetran/painting-classifier-keras-v1

Architecture

• Base Model: MobileNetV2 (pre-trained on ImageNet)
• Approach: Multi-head classification with shared feature extraction
• Framework: TensorFlow/Keras
• Input Size: 224x224 RGB images

Training Details

• Dataset: WikiArt dataset with 84,440 paintings
• Split: 75% training, 25% validation
• Augmentation: Random horizontal flip, rotation, zoom
• Training Stages: Frozen backbone (2 epochs) + full fine-tuning (10 epochs)

Supported Classes

Artists (129 classes)

Claude Monet, Vincent van Gogh, Pablo Picasso, Leonardo da Vinci, Rembrandt, Salvador Dali, Edgar Degas, Pierre-Auguste Renoir, Paul Cezanne, Henri Matisse, Michelangelo, Raphael, Edvard Munch, Gustav Klimt, Wassily Kandinsky, and 114 more.

Genres (11 classes)

Abstract painting, Cityscape, Genre painting, Illustration, Landscape, Nude painting, Portrait, Religious painting, Sketch and study, Still life, Unknown Genre

Art Styles (27 classes)

Impressionism, Post-Impressionism, Cubism, Abstract Expressionism, Early Renaissance, High Renaissance, Northern Renaissance, Mannerism Late Renaissance, Baroque, Rococo, Romanticism, Realism, Expressionism, Symbolism, Art Nouveau, Fauvism, Pointillism, Naive Art Primitivism, Synthetic Cubism, Analytical Cubism, Action painting, Color Field Painting, Contemporary Realism, New Realism, Minimalism, Pop Art, Ukiyo-e

Limitations

• Best performance on Western classical and modern art from the training distribution
• May have reduced accuracy on contemporary art, digital art, or non-Western styles
• "Unknown Artist" and "Unknown Genre" classes may have lower precision
• Requires clear, well-lit images for optimal results

License

MIT License

Links

• Model Repository: https://huggingface.co/meetran/painting-classifier-keras-v1
• Training Dataset: WikiArt on Hugging Face