datomnurdin / tensorflow-cnn

This source code provides example train custom images using Tensorflow.

TensorFlow CNN

This source code provides example train Bird images using Tensorflow.

Steps

Step 1

Install Tensorflow using PIP, https://www.tensorflow.org/install/.

Step 2

Download bird images from Google image using this chrome extension, https://chrome.google.com/webstore/detail/fatkun-batch-download-ima/nnjjahlikiabnchcpehcpkdeckfgnohf?hl=en. Create 12 folders for all birds inside tf_files > birds. Make sure each bird folder contain at least 60-70 images and same category.

Step 3

Use this command to retrain your custom images

sudo python retrain.py --bottleneck_dir=tf_files/bottlenecks --model_dir=tf_files/inception --output_graph=tf_files/retrained_graph.pb --output_labels=tf_files/retrained_labels.txt --image_dir /tf_files/birds

Step 4

Predict

python detect.py sample.png

##FAQ/CONTACT/TROUBLESHOOT Mohammad Nurdin bin Norazan

• http://github.com/datomnurdin
• http://twitter.com/datomnurdin
• [email protected]