This project explores the correlation between movie color schemes and their genres, using image processing and machine learning. It extracts dominant color palettes from movie frames or posters, classifies the genre, and recommends similar movies accordingly.
- 🎨 Extract dominant color schemes from movie posters or frames
- 🧠 Predict movie genres using machine learning models
- 🤖 Recommend similar movies based on color similarity and genre
- 📊 Visualization of color palettes for analysis
- 🗂️ Dataset creation and preprocessing tools
-
Color Scheme Extraction
- Uses
k-means clusteringandsub hull segmentationto identify dominant colors in the entire movie/movie trailer . - Palettes are stored and used as features.
- Uses
-
Genre Classification
- Trained ML model takes color features and predicts the movie genre.
-
Recommendation Engine
- Based on predicted genre and color similarity using cosine similarity and k-NN.
- Clone the repo
- Run the ipynb in your system locally or simply in colab/jupyter.
You can create your own dataset using:
- Public movie poster datasets
- Frame extraction from trailers using ffmpeg
Python,OpenCV , PIL , scikit-learn, matplotlib , seaborn