Я использовал файл .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