在C ++ TensorFlow API中运行Mask RCNN(matterport)训练的模型。但是结果错误

时间:2018-11-13 03:20:08

标签: python c++ tensorflow deep-learning computer-vision

使用以下几行,我可以得到我的output_detections和output_mrcnn_mask

tensorflow::Status run_status = session->Run({ { "input_image", inputTensor },{ "input_image_meta", inputMetadataTensor },{"input_anchors",input_anchors } },
        { "output_detections", "output_mrcnn_class", "output_mrcnn_bbox", "output_mrcnn_mask",
                "output_rois", "output_rpn_class", "output_rpn_bbox" },
        {},
        &outputs);

auto detectionsMap = outputs[0].tensor<float, 3>();
auto y1 = detectionsMap(0, i, 0); float x1 = detectionsMap(0, i, 1); auto y2 = detectionsMap(0, i, 2); float x2 = detectionsMap(0, i, 3) ;

但是,无论我输入了哪张图像,我总是得到以下结果。

  

x1 = x2 = 1.00392

我尝试用Python推断pb文件。除了x1和x2,结果与C ++相同。

环境:

  • Windows 10
  • Visual Studio 2015
  • Python 3.5
  • Python和C ++中的TensorFlow 1.4.0

PS:我使用以下命令为Visual Studio构建TensorFlow: (Build a Tensorflow C++ Library

有人有解决此问题的想法吗?谢谢!!!

0 个答案:

没有答案