TensorFlow模型
这里以本人用TensorFlow实现的AlexNet模型为例,详见文章用TensorFlow实现AlexNet。
且下面的过程都在激活OpenVINO环境后打开的jupyter notebook中完成。
模型转换
编写好AlexNet模型并通过测试后,将模型专为pb文件:
inputs = tf.placeholder(tf.float32, [1, 227, 227, 3])
model = AlexNetModel()
score = model.inference(inputs)
softmax = tf.nn.softmax(score, name='prob')
with tf.Session() as sess:
sess.run(tf.global_variables_initializer())
model.load_original_weights(sess)
constant_graph = tf.graph_util.convert_variables_to_constants(sess, sess.graph_def,