4

我想将预先存在的张量流图转换为可以部署在 Movidius 神经计算棒上的图。

当我使用命令 mvNCCompile（来自 NCSDK）时，出现以下错误：[错误 5] 工具包错误：不支持阶段详细信息：不支持动态输入形状。

检查张量板中的张量流图后，我发现输入张量具有以下形状：(?,?,?,3) 在我的用例中，我始终具有相同的输入形状：(1,368,656,3)。

现在我想将张量的动态输入形状更改为静态输入形状。

我怎样才能做到这一点？

我试过关注这篇文章：如何在 Tensorflow 中更改保存的模型输入形状？ 但我无法更改图表。

图表可以在这里找到：https ://github.com/ildoonet/tf-pose-estimation/blob/master/models/graph/mobilenet_thin/graph_opt.pb

import tensorflow as tf

graph_path = 
'/home/bk/Documents/OPSLim/Pose/graph_models/mobilenet_thin/graph_opt.pb'

with tf.gfile.GFile(graph_path, 'rb') as f:
    graph_def = tf.GraphDef()
    graph_def.ParseFromString(f.read())


graph = tf.get_default_graph()
tf_new_image = tf.placeholder(shape=(1, 368, 368, 3), dtype='float32', 
name='new_image')
tf.import_graph_def(graph_def, name='TfPoseEstimator', input_map= {"image:0": tf_new_image})
4

0 回答 0