關於我自己

我的相片
New York, New York, United States
我叫江奕賢啦

2018年3月10日

15分鐘教你如何訓練電腦看照片 (例: 辨識花的種類)


我是江博士 今天花十五分鐘教大家怎麼用 tensorflow 訓練電腦看照片區別不同的花
學會之後 你只需要提供不同的照片給他 他就會辨識不同的東西
看你要辨識植物、花卉、鳥類、醫學影像、衛星影像、、、都行。
完全不用寫程式喔!

#準備
#在 /f/Docker/tensorflow 裡面準備好 images 目錄, 裡面分目錄放照片(目錄名稱用英文)

#進 docker
cd /f/Docker/tensorflow
docker run -it -v /f/Docker/tensorflow:/tf_files gcr.io/tensorflow/tensorflow bash

#在 docker 內
cd /tf_files
curl -O https://raw.githubusercontent.com/tensorflow/tensorflow/master/tensorflow/examples/image_retraining/retrain.py
curl -O https://raw.githubusercontent.com/tensorflow/tensorflow/master/tensorflow/examples/label_image/label_image.py

#訓練 (會自動download inception)
python retrain.py --bottleneck_dir=/tf_files/bottlenecks --how_many_training_steps 1000 --model_dir=/tf_files/inception --output_graph=/tf_files/retrained_graph.pb --output_labels=/tf_files/retrained_labels.txt --image_dir /tf_files/images

#辨識
python label_image.py --image test/GYY.jpg --labels ./retrained_labels.txt --graph ./retrained_graph.pb --input_layer Mul --output_layer final_result
python label_image.py --image test/ChiangWei.jpg --labels ./retrained_labels.txt --graph ./retrained_graph.pb --input_layer Mul --output_layer final_result
python label_image.py --image test/rose.jpg --labels ./retrained_labels.txt --graph ./retrained_graph.pb --input_layer Mul --output_layer final_result