小编Анд*_*нко的帖子

使用 OpenCV C++ 进行 Tensorflow 2 对象检测

我已经使用 Tensorflow 2 对象检测 API 训练了 SSD ResNet V1 模型。然后我想在 C++ 代码中将这个模型与 OpenCV 一起使用。

首先,训练后我有三个文件:

  • 检查站
  • ckpt-101.data-00000-of-00001
  • ckpt-101.index

请注意,我没有 .meta 文件,因为它不是生成的。

然后我使用exporter_main_v2.py对象检测 API 中的脚本从这些文件创建了 SavedModel :

python3 exporter_main_v2.py input_type=image_tensor --pipeline_config_path /path/to/pipeline.config --trained_checkpoint_dir=/path/to/checkouts --output_directory=/path/to/output/directory
Run Code Online (Sandbox Code Playgroud)

运行此脚本后,我得到了 saved_model.pb

我试图以这种方式在 OpenCV 中使用这个文件:

cv::dnn::Net net = cv::dnn::readNetFromTensorflow("/path/to/saved_model.pb");
Run Code Online (Sandbox Code Playgroud)

但我收到以下错误:

OpenCV(4.2.0) /home/andrew/opencv/modules/dnn/src/tensorflow/tf_io.cpp:42: error: (-2:Unspecified error) FAILED: ReadProtoFromBinaryFile(param_file, param). Failed to parse GraphDef file: /home/andrew/Documents/tensorflow_detection/workspace/pb_model/saved_model/saved_model.pb in function 'ReadTFNetParamsFromBinaryFileOrDie'
Run Code Online (Sandbox Code Playgroud)

然后我尝试冻结saved_model.pb。但是,据我所知,在 TF2.x 中这是不可能的,因为 TF2.x 不支持会话和图形。我也没有 .pbtxt 文件。

我的问题:是否可以在 OpenCV C++ 中使用通过 TF2 Object Detection API 训练的模型?

如果您帮助我解决此问题或提供任何有用的建议,我将不胜感激。

c++ opencv resnet object-detection-api tensorflow2.0

7
推荐指数
1
解决办法
2593
查看次数