tf11: retrain谷歌Inception模型

 

 

前一帖使用到了谷歌训练的Inception模型，本帖就基于Inception模型retrain一个图像分类器。

图像分类器应用广泛，连农业都在使用，如判断黄瓜种类。

本帖使用的训练数据是PixelCNN一文中使用的妹子图，最后训练出的分类器可以判断图片是不是妹子图。

首先下载tensorflow源代码：

 

$ git clone https://github.com/tensorflow/tensorflow
$ git checkout r0.11 

 

 

使用examples中的image_retraining。

训练：

 

$ python tensorflow/tensorflow/examples/image_retraining/retrain.py --bottleneck_dir bottleneck --how_many_training_steps 4000 --model_dir model --output_graph output_graph.pb --output_labels output_labels.txt