网上关于这个的博文很多,作为一个新手想要弄明白并且实现,还需要折腾一番。以下是我的实现过程,希望对像我一样的新手有帮助!
步骤:
1、下载google的inception-v3模型,链接:http://download.tensorflow.org/models/image/imagenet/inception-2015-12-05.tgz
2、准备自己的训练图片:我建立了一个images文件夹,在里面又建立了5个子文件夹(cats、dogs、planes、houses、guitars)
注意:此处一定要像我一样分类建立好文件夹,然后将训练的图片放入相应的文件夹中
3、进入github,下载的tensorflow(https://github.com/tensorflow/tensorflow)里image_retraining文件里面的retrain.py文件,此具体路径:tensorflow——examples——image_retraining——retrain.py。但是,可能有人按这个路径进入后发现没有这个文件,如下图:
原因是tensorflow的版本问题。tensorflow1.7版本及以上会安装tensorflow hub来获取bottleneck的输出,训练最后最后一层;以前的版本是采用冻结bottleneck层之前的参数,来得到