Skip to content

Instantly share code, notes, and snippets.

@IAMIronmanSam
Created October 7, 2016 21:40
Show Gist options
  • Save IAMIronmanSam/073a03997464cf64e852141ea756a74c to your computer and use it in GitHub Desktop.
Save IAMIronmanSam/073a03997464cf64e852141ea756a74c to your computer and use it in GitHub Desktop.
Image Classifier
#Deplo tensorflow image in docker
sudo docker run -it gcr.io/tensorflow/tensorflow:latest-devel
#Copy Image Set-
sudo docker run -it -v /home/arivu/MyWorld/tensorflow/tf_files/starwars:/tf_files/starwars gcr.io/tensorflow/tensorflow:latest-devel
#Re-Train model
python /tensorflow/tensorflow/examples/image_retraining/retrain.py \
--bottleneck_dir=/tf_files/bottlenecks \
--how_many_training_steps 500 \
--model_dir=/tf_files/inception \
--output_graph=/tf_files/retrained_graph.pb \
--output_labels=/tf_files/retrained_labels.txt \
--image_dir /tf_files/starwars
@AhmetCanSolak
Copy link

I have the same problem with you. Were you able to find a solution?

@Syirrus
Copy link

Syirrus commented Apr 18, 2017

I'm having this problem too

@jingfengli
Copy link

You should link the '/home/arivu/MyWorld/tensorflow' not just '/home/arivu/MyWorld/tensorflow/starwars'. Otherwise, the train model will be deleted after you exit the docker session.
sudo docker run -it -v /home/arivu/MyWorld/tensorflow/tf_files:/tf_files gcr.io/tensorflow/tensorflow:latest-devel

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment