Как изменить форму ввода существующей модели тензорного потока (.pb)
Я хочу превратить существующий граф тензорного потока в граф, который можно развернуть на ручке Movidius Neural Compute.
Когда я использую команду 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})