MobileNet训练分类网

MobileNet训练分类网

1.环境准备

• 安装tensorflow：

  pip install tensorflow

• 验证安装成功：

  python
  import tensorflow as tf
  tf.__version__

• 把 tensorflow/examples/git clone下来

• ​

2.数据集文件结构

• 数据

  ../example/image_retraining/ :
  --dataset
  --class1
    --image1.jpg
    --image2.jpg
    ...
  --class2
    --image1.jpg
    ...
  ...  

• 标签

  ../label.txt :
  class1
  class2
  ...

  ​

3.训练脚本

• ../example/image_retraining/retrain.py 的主函数中有各项训练参数的设置，训练命令例如：

  python retrain.py \
    --image_dir /home/wz/TensorFlow/tensorflow/tensorflow/examples/image_retraining/dataset \
    --intermediate_store_frequency 1000 \
    --how_many_training_steps 50000 \
    --learning_rate 0.001 \
    --train_batch_size 8 \ 
    --testing_percentage 0 \
    --eval_step_interval 20 \
    --architecture mobilenet_1.0_224 \
    --output_graph  /home/wz/TensorFlow/tensorflow/tensorflow/examples/image_retraining/tmp/output_graph.pb \
    --intermediate_output_graphs_dir /home/wz/TensorFlow/tensorflow/tensorflow/examples/image_retraining/tmp/intermediate_graph \
    --output_labe