Как изменить форму ввода существующей модели тензорного потока (.pb) - PullRequest
Новая
       92

Как изменить форму ввода существующей модели тензорного потока (.pb)

1 голос
/ 30 мая 2019

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