20170512-110547.pb 可以转换为 opencv可以使用的模型吗
时间: 2023-11-27 13:55:14 浏览: 65
pb模型转pbtxt,opencv调用
4星 · 用户满意度95%
20170512-110547.pb 是 TensorFlow 的模型文件,如果你想在 OpenCV 中使用它,需要将其转换为 OpenCV 可以使用的格式。你可以使用 TensorFlow 的 freeze_graph 工具将其转换为 .pbtxt 格式,然后使用 OpenCV 的 DNN 模块加载它。具体操作步骤如下:
1. 安装 TensorFlow 和 OpenCV 库。
2. 在终端中运行以下命令,将 TensorFlow 的模型文件 .pb 转换为 .pbtxt 文件:
```
python -m tensorflow.python.tools.freeze_graph \
--input_graph=/path/to/20170512-110547.pb \
--output_graph=/path/to/20170512-110547.pbtxt \
--input_checkpoint=/path/to/20170512-110547.ckpt \
--output_node_names=Softmax
```
其中,--input_graph 指定输入的 TensorFlow 模型文件,--output_graph 指定输出的 .pbtxt 文件,--input_checkpoint 指定原始模型的参数文件,--output_node_names 指定输出的节点名称。
3. 使用以下代码加载 .pbtxt 文件:
```
cv::dnn::Net net = cv::dnn::readNetFromTensorflow("/path/to/20170512-110547.pbtxt");
```
这样就可以使用 OpenCV 的 DNN 模块加载 TensorFlow 的模型了。
阅读全文