Add Pluto and other Dwarf planets about celestial-bodies-detection HOT 9 OPEN

Hey @ritwik12 , I'm interested in contributing to this issue, so before I start working on it, would you mind sparing your time explaining what the issue is about and pointing me to some resources to get started.

from celestial-bodies-detection.

Yeah. so just like we have 8 planets right now for the classification. Our model is not trained for Pluto and Dwarf planets. If we try to give an image of Pluto to our model. It will classify it as wrong.

from celestial-bodies-detection.

Need to add Training images for Pluto, then add test images. Train the model based on those images.

from celestial-bodies-detection.

Hello @ritwik12 ,

I'm interested in solving this issue, and I cloned this repository and I collected some pluto images and added them to both training and test data and added "pluto" to retrained_labels.txt and label_image.py and now how to start training ??is this the correct way ??

from celestial-bodies-detection.

@SSahas Please follow the readme.md. I think you also don't need to update retrained_labels.txt as that is something done on its own with this command below.

To train model you need to run python retrain.py --bottleneck_dir=bottlenecks --how_many_training_steps=500 --model_dir=inception --summaries_dir=training_summaries/basic --output_graph=retrained_graph.pb --output_labels=retrained_labels.txt --image_dir=./training_data

from celestial-bodies-detection.

hey @ritwik12 , I trained the model for classifying Pluto planet and got the final test accuracy : 87% , I also run the run.py and the webapp is classifying Pluto images correctly but the retrained_graph.pb file is greater than 50 mb so I am not able to push it , can I use git LFS now??

from celestial-bodies-detection.

@SSahas You meant retrained_graph.pb? The current one in repo is already 83MB

from celestial-bodies-detection.

hey @ritwik12 ,
I opened a pull request called "Train the model for detecting pluto".
I trained the model and the flask app can also able to detect the images as pluto .
please look in to it.

please tell if there are any things to be changed.

from celestial-bodies-detection.

Hey @ritwik12 , I'm interested in contributing to this issue. This is my first time contributing. I want to add a dwarf planet named 'Makemake' to the classification.

from celestial-bodies-detection.