我正在尝试加载以下张量流模型https://github.com/NVIDIA-AI-IOT/redtail/tree/master/stereoDNN/models/ResNet-18_2D/TensorFlow
我是从 google colab 做的,这就是我所做的:
%tensorflow_version 1.x
!git clone https://github.com/NVIDIA-AI-IOT/redtail
with tf.Session() as sess:
new_saver = tf.train.import_meta_graph('./redtail/stereoDNN/models/ResNet-18_2D/TensorFlow/model-inference-513x257-0.meta')
new_saver.restore(sess, tf.train.latest_checkpoint('./redtail/stereoDNN/models/ResNet-18_2D/TensorFlow/'))
print(sess.run('w1:0'))
运行时抛出以下错误restore
:
ValueError:当它为 None 时无法加载 save_path。
根据互联网搜索,这是由于文件夹中没有检查点文件。这些是加载模型所必需的吗?在这种情况下我应该如何加载模型?