#python #tensorflow #opencv #keras
Вопрос:
Как говорится в названии, мне нужно получить имена выходных узлов из моей модели keras xception. В частности, я хочу использовать эту информацию, чтобы заморозить ее для использования с opencv.
Комментарии:
1. Привет, добро пожаловать в SO! Не могли бы вы, пожалуйста, сформулировать это более четко? Хотели бы вы получить вес вашей модели после эпохи?
2. Я пытаюсь получить имена выходных данных для использования freeze_graph.freeze_graph.
Ответ №1:
Попробуй это.
model = keras.models.load_model( 'yourModel.h5' ) output = model.output выход:
KerasTensor(type_spec=TensorSpec(форма=(Нет, 128), dtype=tf.float32, имя=Нет), имя=»Bottleneck_BatchNorm/batchnorm/add_1:0″, описание=»создано слоем «Bottleneck_BatchNorm»»)
Комментарии:
1. Я попробовал это, и я получаю эту ошибку: ошибка утверждения: dense/BiasAdd:0 отсутствует в графике (вывод dense/BiasAdd:0). Полное сообщение: KerasTensor(type_spec=TensorSpec(форма=(Нет, 1), dtype=tf.float32, имя=Нет), имя=»плотный/смещение:0″, описание=»создан слоем «плотный»»)
2. Также в качестве примечания я получаю два файла .pb из model.save(каталог model_). Я предполагаю, что напрямую использовать их невозможно.