首页 > 解决方案 > Serialize a graph with phase_train 变成 false 是什么意思?

问题描述

我想在 opencv 中导入一个 TensorFlow 模型(.pb 文件)。

import cv2 
cv2.dnn.readNetFromTensorflow("./mars-small128.pb")

但这总是给我这个错误:

opencv/modules/dnn/src/tensorflow/tf_importer.cpp:497: error: (-2:Unspecified error) Input layer not found: Const in function 'connect'

我在谷歌上搜索了很长时间,但没有成功解决这个错误。但是这个线程链接说:

请序列化一个将 phase_train 设置为 false 的图。

但我不知道该怎么做。我正在探索深度排序,他们提供了这些文件(.ckpt、.ckpt.meta 和 .pb 文件)以及用于重新生成 .pb 文件的 python脚本
那么谁能告诉我我需要在哪里传递那个参数phase_train或者你是否有任何其他解决方案来解决这个错误?
OpenCv 版本 -:3.4.2 和 4.1.0(在两个版本上都试过)
PS。:如果有人可以解释或提供任何好的链接来解释序列化图形的含义,那就太好了。

标签: pythonopencvtensorflow

解决方案


推荐阅读