用inception-v3模型,训练最后一层,打造自己的模型

网上关于这个的博文很多,作为一个新手想要弄明白并且实现,还需要折腾一番。以下是我的实现过程,希望对像我一样的新手有帮助!

步骤:

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层之前的参数,来得到

评论 8
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值