问题解决转载博客： https://blog.csdn.net/Daker_Huang/article/details/90212072
只需要改第三行（源pb文件）和倒数第二行的路径（输出pb文件）
import tensorflow as tf from tensorflow.tools.graph_transforms import TransformGraph with tf.gfile.FastGFile('frozen_inference_graph.pb', 'rb') as f: graph_def = tf.GraphDef() graph_def.ParseFromString(f.read()) graph_def = TransformGraph(graph_def, ['image_tensor'], ['detection_boxes', 'detection_classes', 'detection_scores', 'num_detections'], ['sort_by_execution_order']) with tf.gfile.FastGFile('sorted_inference_graph.pb', 'wb') as f: f.write(graph_def.SerializeToString())
