Замораживание тонко настроенного графика для TensorFlowSharp с TF 1.4 - PullRequest
Новая
       26

Замораживание тонко настроенного графика для TensorFlowSharp с TF 1.4

0 голосов
/ 19 ноября 2018

Я настроил модель (используя TF 1.9) из Object Zoo Model , и сейчас я пытаюсь заморозить график для TensorFlowSharp, используя TF 1.9.

import tensorflow as tf
import os
from tensorflow.python.tools import freeze_graph
from tensorflow.core.protobuf import saver_pb2

#print("current tensorflow version: ", tf.version)
sess=tf.Session()

model_path = 'latest_cp/'
saver = tf.train.import_meta_graph('model.ckpt.meta')
saver.restore(sess,tf.train.latest_checkpoint('.')) #current dir of the checkpoint file
tf.train.write_graph(sess.graph_def, '.', 'test.pbtxt') #output in pbtxt format

freeze_graph.freeze_graph(input_graph = 'test.pbtxt',
    input_binary = False,
    input_checkpoint = model_path + 'model.ckpt',
    output_node_names =     "num_detections,detection_boxes,detection_scores,detection_classes",
    output_graph = 'test.bytes' ,
    clear_devices = True, initializer_nodes = "",input_saver = "",
    restore_op_name = "save/restore_all", filename_tensor_name = "save/Const:0")

Это сработало, но после того, как я импортировал его в Unity, он вернул следующую ошибку:

TFException: Op type not registered 'NonMaxSuppressionV3' in binary running on AK38713. Make sure the Op and Kernel are registered in the binary running in this process.

Я обнаружил, что TensorFlowSharp работает с TensorFlow 1.4 икогда я пытался заморозить график с 1.4, он возвращает ту же ошибку NonMaxSuppressionV3.Знаете ли вы какой-нибудь способ решить эту проблему?Большое спасибо за поддержку.

...