Celestial Bodies Detection
CNN-based classification of celestial bodies
A Flask web app that classifies images of planets, moons, asteroids and galaxies with a CNN, then pulls back facts about the result from Wikipedia.
What it does
- Trained a Convolutional Neural Network with TensorFlow and Keras to classify celestial bodies across 12 classes, from asteroids to spiral galaxies.
- Built a Flask front end where users upload an image or paste a URL and get a prediction with confidence scores for every class.
- Enriched predictions by pulling relevant facts and statistics from the Wikipedia API, plus a reverse image search shortcut to learn more.