树莓派3B运行TensorFlow MNIST模型时恢复路径报错求助
解决树莓派上TensorFlow模型恢复的路径错误问题
看起来你遇到的是跨平台路径解析差异导致的小坑,我之前在树莓派上部署TensorFlow模型时也碰到过类似问题,给你几个可行的解决思路:
1. 检查并修复checkpoint文件内容
错误提示里的.model.ckpt非常可疑,大概率是自动生成的checkpoint文件里的路径被意外修改了。打开和脚本同目录的checkpoint文件,你会看到类似这样的内容:
model_checkpoint_path: ".model.ckpt" all_model_checkpoint_paths: ".model.ckpt"
把里面的.model.ckpt全部改成model.ckpt,保存后再运行脚本试试。这个文件是TensorFlow训练时自动生成的,跨系统迁移时可能因为路径规则差异出现格式错误。
2. 使用绝对路径加载模型
相对路径在不同系统的解析逻辑有细微差别,尤其是树莓派的Linux环境和Windows的路径规则不一样。你可以用Python获取脚本的绝对路径,再拼接模型文件名,彻底避免相对路径的解析问题:
import os # 获取当前脚本所在的绝对目录 current_dir = os.path.dirname(os.path.abspath(__file__)) # 拼接模型的绝对路径 model_full_path = os.path.join(current_dir, "model.ckpt") with tf.Session() as sess: sess.run(init_op) saver.restore(sess, model_full_path) prediction=tf.argmax(y_conv,1) toReturn = [] for value in imvalue: toReturn.append(prediction.eval(feed_dict={x: [value],keep_prob: 1.0}, session=sess))
3. 确认文件读取权限
树莓派的Linux系统对文件权限要求更严格,你可以先检查模型文件的权限,确保脚本有读取权限:
在终端执行命令查看权限:
ls -l
如果模型文件的权限没有r(可读)标识,用下面的命令修改权限:
chmod +r model.ckpt* checkpoint
这些方法应该能解决你的问题,毕竟模型在Windows和Ubuntu都能正常运行,说明模型本身没问题,就是树莓派上的路径或权限小细节没处理好。
内容的提问来源于stack exchange,提问作者Marco D.G.
相关产品推荐
相关产品推荐

