TensorFlow-Slim 圖片分類(二)
阿新 • • 發佈:2019-02-15
TensorFlow-Slim圖片分類 TensorFlow-Slim image classification
--dataset_name=flowers \
--dataset_dir=H:\pythoncode\Tensorflow\ImageClassify\data\flowers\record
--train_dir=H:\pythoncode\Tensorflow\ImageClassify\data\flowers\train_log \
--dataset_name=flowers \
--dataset_split_name=train \
--dataset_dir=H:\pythoncode\Tensorflow\ImageClassify\data\flowers\record \
--model_name=inception_v3
python train_image_classifier.py \
--train_dir=H:\pythoncode\Tensorflow\ImageClassify\data\flowers\train_log \
--dataset_dir=H:\pythoncode\Tensorflow\ImageClassify\data\flowers\record \
--dataset_name=flowers \
--dataset_split_name=train \
--model_name=inception_v3 \
--checkpoint_path=H:\pythoncode\Tensorflow\ImageClassify\pretrained\inception_v3.ckpt \
--checkpoint_exclude_scopes=InceptionV3/Logits,InceptionV3/AuxLogits \
--trainable_scopes=InceptionV3/Logits,InceptionV3/AuxLogits
tensorboard --logdir=H:\pythoncode\Tensorflow\ImageClassify\data\flowers\train_log
--checkpoint_path=H:\pythoncode\Tensorflow\ImageClassify\data\flowers\train_log \
--eval_dir=H:\pythoncode\Tensorflow\ImageClassify\data\flowers\validation_log \
--dataset_name=flowers \
--dataset_split_name=validation \
--dataset_dir=H:\pythoncode\Tensorflow\ImageClassify\data\flowers\record \
--model_name=inception_v3
python export_inference_graph.py \
--alsologtostderr \
--model_name=inception_v3 \
--output_file=H:\pythoncode\Tensorflow\ImageClassify\data\flowers\inception_v3_inf_graph.pb
--dataset_name flowers
--input_graph H:\pythoncode\Tensorflow\ImageClassify\data\flowers\inception_v3_inf_graph.pb \
--input_checkpoint H:\pythoncode\Tensorflow\ImageClassify\data\flowers\train_log\model.ckpt-12780 \
--input_binary true \
--output_node_names InceptionV3/Predictions/Reshape_1 \
--output_graph H:\pythoncode\Tensorflow\ImageClassify\data\flowers\frozen_graph.pb
--model_path H:\pythoncode\Tensorflow\ImageClassify\data\flowers\frozen_graph.pb \
--label_path H:\pythoncode\Tensorflow\ImageClassify\data\flowers\record\labels.txt \
--image_file dandelion.jpg
參考資料
[1] https://github.com/tensorflow/models/tree/master/research/slim
1 將圖片資料轉換成TF-Record格式檔案 Converting Images to TFRecord format
python download_and_convert_data.py \--dataset_name=flowers \
--dataset_dir=H:\pythoncode\Tensorflow\ImageClassify\data\flowers\record
2 從頭開始訓練 Training a model from scratch
python train_image_classifier.py \--train_dir=H:\pythoncode\Tensorflow\ImageClassify\data\flowers\train_log \
--dataset_name=flowers \
--dataset_split_name=train \
--dataset_dir=H:\pythoncode\Tensorflow\ImageClassify\data\flowers\record \
--model_name=inception_v3
3 預訓練模型調優 Fine-tuning a model from an existing checkpoint
--train_dir=H:\pythoncode\Tensorflow\ImageClassify\data\flowers\train_log \
--dataset_dir=H:\pythoncode\Tensorflow\ImageClassify\data\flowers\record \
--dataset_name=flowers \
--dataset_split_name=train \
--model_name=inception_v3 \
--checkpoint_path=H:\pythoncode\Tensorflow\ImageClassify\pretrained\inception_v3.ckpt \
--checkpoint_exclude_scopes=InceptionV3/Logits,InceptionV3/AuxLogits \
--trainable_scopes=InceptionV3/Logits,InceptionV3/AuxLogits
4 TensorBoard
5 模型能力評估 Evaluating performance of a model
python eval_image_classifier.py \--checkpoint_path=H:\pythoncode\Tensorflow\ImageClassify\data\flowers\train_log \
--eval_dir=H:\pythoncode\Tensorflow\ImageClassify\data\flowers\validation_log \
--dataset_name=flowers \
--dataset_split_name=validation \
--dataset_dir=H:\pythoncode\Tensorflow\ImageClassify\data\flowers\record \
--model_name=inception_v3
6 匯出推理圖 Exporting the Inference Graph
--alsologtostderr \
--model_name=inception_v3 \
--output_file=H:\pythoncode\Tensorflow\ImageClassify\data\flowers\inception_v3_inf_graph.pb
--dataset_name flowers
7 匯出權重 Freezing the exported Graph
python freeze_graph.py \--input_graph H:\pythoncode\Tensorflow\ImageClassify\data\flowers\inception_v3_inf_graph.pb \
--input_checkpoint H:\pythoncode\Tensorflow\ImageClassify\data\flowers\train_log\model.ckpt-12780 \
--input_binary true \
--output_node_names InceptionV3/Predictions/Reshape_1 \
--output_graph H:\pythoncode\Tensorflow\ImageClassify\data\flowers\frozen_graph.pb
8 預測 Test
python classify_image_inception_v3.py \--model_path H:\pythoncode\Tensorflow\ImageClassify\data\flowers\frozen_graph.pb \
--label_path H:\pythoncode\Tensorflow\ImageClassify\data\flowers\record\labels.txt \
--image_file dandelion.jpg
參考資料
[1] https://github.com/tensorflow/models/tree/master/research/slim