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

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

你可能感兴趣的:(量化)