1、写.pb文件:
graph_def = tf.get_default_graph().as_graph_def()#将图定义取出
out_graph = graph_util.convert_variables_to_constants(sess, graph_def,output_node_names=[‘output’])#将图中的变量转化为constant
with tf.gfile.GFile(’./model.pb’,‘wb’) as f:
f.write(out_graph.SerializeToString())#将图定义转化为字符串形式并且写入.pb文件中
python模型到apk
最新推荐文章于 2024-05-28 04:00:00 发布