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

最新推荐文章于 2023-09-12 17:32:07 发布

风极_陨

最新推荐文章于 2023-09-12 17:32:07 发布

阅读量1.8k

点赞数 3

分类专栏： OpenCv-C++学习记录文章标签：迁移学习模型转化问题 assert(graph_def.node[0].op == &#039;Pla

本文链接：https://blog.csdn.net/Daker_Huang/article/details/90212072

版权

OpenCv-C++学习记录专栏收录该内容

56 篇文章 43 订阅

订阅专栏

在迁移学习（使用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命令进行转化即可。
结果如图：
在这里插入图片描述