slim框架下的模型的运行及保存成.pb文件

参考:https://blog.csdn.net/lwplwf/article/details/76177296
1、将数据集下载到
在这里插入图片描述
2、下载代码,并进入slim文件中
运行convert_tfrecord.sh脚本,将数据集转成成tfrecord格式。
3、按照参考博客运行run_train.sh脚本。出现如下错误(目前猜测可能是需要gpu,而本人太穷,还是用自己的破电脑运行)
希望知道解决方法的大佬分享一下方法。
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
上面的问题确实是gpu问题,在源代码中改了一个参数,运行顺利,不过还是有问题。
在这里插入图片描述
这次问题估计还是我的电脑配置低的原因。果然cpu还是不能跑神经网络啊

小白总结一下:inception_v4模型的checkpoint文件:文件保存了一个目录下所有的模型文件列表。解压之后是.ckpt文件。
TensorFlow模型会保存在后缀为.ckpt的文件中。保存后在save这个文件夹中实际会出现3个文件,因为TensorFlow会将计算图的结构和图上参数取值分开保存。这三个文件分别是:
model.ckpt.meta文件保存了TensorFlow计算图的结构,可以理解为神经网络的网络结构
model.ckp.datat文件保存了TensorFlow程序中每一个变量的取值
model.ckp.index:每个变量的索引(我猜一个就是网络连接的意思吧)
checkpoint文件保存了一个目录下所有的模型文件列表。
另外.pb文件:训练好的模型保存形式。
参考:https://blog.csdn.net/u014264373/article/details/79943389
https://blog.csdn.net/suixinsuiyuan33/article/details/68953766
https://www.zhihu.com/question/61946760

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 8
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 8
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值