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

本文详细介绍了如何使用Inception_v4模型进行图像识别任务的训练流程,包括数据集准备、模型转换、运行训练脚本及常见问题解决。特别讨论了GPU需求与配置问题,以及模型文件的保存格式,如.ckpt、.meta、.data和.index文件的作用。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

参考:https://blog.youkuaiyun.com/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.youkuaiyun.com/u014264373/article/details/79943389
https://blog.youkuaiyun.com/suixinsuiyuan33/article/details/68953766
https://www.zhihu.com/question/61946760

评论 8
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值