如何将faster_rcnn模型加载到Opencv?

时间:2018-06-27 15:49:09

标签: python opencv tensorflow object-detection object-detection-api

我一直在尝试将Tensorflow对象检测模型Zoo中的faster_rcnn_resnet101_coco_2018_01_28模型加载到OpenCV中,并且遇到了麻烦。我使用位于OpenCV最新版本的samples \ dnn目录中的tf_text_graph_faster_rcnn.py脚本来生成OpenCV所需的.pbtxt文件。然后,我使用readNetFromTensorflow(.pb,.pbtxt)(。pb是faster_rcnn_resnet101_coco_2018_01_28的Frozen_inference_graph.pb文件,而.pbtxt是tf_text_graph_graph_faster_rcnn.py由Frozen_inference_graph.pb生成的文件),并按照以下说明进行接收:

Exception: OpenCV(3.4.1-dev) d:\externallibraries\opencv\opencv- 
3.4\modules\dnn\src\tensorflow\tf_importer.cpp:484: error: (-2:Unspecified 
error) Const kernel input not found in function 
'cv::dnn::experimental_dnn_v4::`anonymousnamespace'::TFImporter::getConstBlob'

我尝试在ssd类型的模型中加载而没有任何问题,但是fast_rcnn类型的模型始终会抛出相同的错误。对于解决此问题的任何帮助将不胜感激。

谢谢

0 个答案:

没有答案