使用以下几行,我可以得到我的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 ++相同。
环境:
PS:我使用以下命令为Visual Studio构建TensorFlow: (Build a Tensorflow C++ Library)
有人有解决此问题的想法吗?谢谢!!!