opencv dnn模块加载tensorflow.pb文件错误

Question

我用tensorflow2.4训练了一个mnist_fashion模型，然后使用opencv调用生成的.pb文件，出现了以下错误。

Net net = readNetFromTensorflow(weightFile);

解析协议缓冲区时，字符串字段"tensorflow.FunctionDef.Node.ret"包含无效的UTF-8数据。如果要发送原始字节，请使用"字节"类型。无法分析cv:：dnn:：ReadTFNetParamsFromBinaryFileOrDie 中的GraphDef文件：models/saved_model.pb(

Answer 1

我找到了一个解决方案，只需将saved_model.pb转换为frozen_graph.pb。[https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/tools/freeze_graph.py][1]