opencv dnn 模塊加載 tensorflow.pb 文件錯誤

Question

我用tensorflow2.4訓練了一個mnist_fashion model，然后用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]