Как загрузить график из файла .ckpt - PullRequest
0 голосов
/ 30 января 2019

Я использовал файл .pb для загрузки графика при обучении модели inception v3 с использованием этого кода

with tf.Graph().as_default() as graph:
    model_path = os.path.join(MODEL_DIR, model_info['model_file_name'])
    print('Model path: ', model_path)
    with gfile.FastGFile(model_path, 'rb') as f:
        graph_def = tf.GraphDef()
        graph_def.ParseFromString(f.read())
        bottleneck_tensor, resized_input_tensor = (tf.import_graph_def(graph_def, name='', return_elements=[model_info['bottleneck_tensor_name'], model_info['resized_input_tensor_name'],]))

, где model_info - это словарь

data_url = 'http://download.tensorflow.org/models/image/imagenet/inception-2015-12-05.tgz'    
bottleneck_tensor_name = 'pool_3/_reshape:0'
bottleneck_tensor_size = 2048
input_width = 299
input_height = 299
input_depth = 3
resized_input_tensor_name = 'Mul:0'
model_file_name = 'classify_image_graph_def.pb'
input_mean = 128
input_std = 128

, теперь я хочуиспользуйте Inception v4, и он сохраняется в файле .ckpt.как загрузить график для соответствия этому формату, используя файл .ckpt

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...