Как изменить форму входного тензора существующего графа (например, MobileNet) в Tensorflow?
Я загружаю и прогнозирую с помощью MobileNet v1 frozen_graph.pb и работаю с ним в Tensorflow без проблем. Для тех, кто знаком с MobileNet Input:0 это [?, 224, 224, 3]. Я пытаюсь оптимизировать аппаратный ускоритель, который может пакетировать до 16, поэтому мне нужно изменить input:0 до [16, 224, 224, 3].
Итак, очевидно, я могу создать новый тензор с правильной формой:
:
input_tensor = graph.get_tensor_by_name('input:0')
new_tensor = tf.reshape(input_tensor, [16, 224, 224, 3], name='input')
:
Это создает новый тензор с именем "input_1", поскольку "input" уже существует.
Как мне на самом деле заменить этот "входной" тензор в загруженном в данный момент графе новым "input_1", чтобы он имел входной тензор [16, 224, 224, 3]? Я знаю, как выписать модель замораживания и все такое, просто не знаю, как обрезать / привить.
Заранее спасибо.