Я загружаю и прогнозирую с помощью 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]?Я знаю, как написать модель замораживания и все такое, просто не знаю, как обрезать / привить.
Заранее спасибо.