Machine learning · Computer vision
Image Popularity Prediction
An end-to-end computer-vision pipeline that predicts image popularity using object detections, spatial features, and color statistics.
- Processed roughly 20,000 images with YOLOv11
- Compared Random Forest and LightGBM models
- Deployed an interactive Streamlit application