使用Opencv在基于SSD-MobileNet迁移学习中生成pbtxt文件遇到的问题及解决方案

在迁移学习（使用MobileNet）训练后，如果要是用opencv调用pb模型，需要的“佐料”是pbtxt文件，这个文件必不可少。
我在使用opencv4.0.0中的tf_text_graph_ssd.py（文件路径在“D:\opencv\opencv_4.0.0\sources\samples\dnn”）生成pbtxt文件时，遇到了这样一个问题：

Traceback (most recent call last):
  File "tf_text_graph_ssd.py", line 276, in <module>
    createSSDGraph(args.input, args.config, args.output)
  File "tf_text_graph_ssd.py", line 129, in createSSDGraph
    assert(graph_def.node[0].op == 'Placeholder')
AssertionError

如下图：

这时候的解决方案则是把frozen_inference_graph.pb转换一下，使用python程序：

import tensorflow as tf
from tensorflow.tools.graph_transforms import TransformGraph

with tf.gfile.FastGFile('F:/live/face_data/pb_model/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('F:/live/face_data/pb_model/sorted_inference_graph.pb', 'wb') as f:
        f.write(graph_def.SerializeToString())

然后重新调用cmd命令进行转化即可。
结果如图：