将 Mask-RCNN (Tensorflow) 导出到 C++

如何解决将 Mask-RCNN (Tensorflow) 导出到 C++

我是这个平台的新手，任何帮助或帮助将不胜感激。

我正在执行一项任务，即导出最初在 Python 中训练的 Mask-RCNN 模型（为岩石检测而训练），以用于我们的机器人框架（基于 C++ 和 Java）。为此，我有经过训练的权重文件（我从 .h5 转换为 .pb）但是当我尝试运行 python 脚本（只是为了可视化该 .pb 文件）时，我得到了一些我真的无法理解的东西我真正想要的是边界框坐标或掩码坐标。

x = graph.get_tensor_by_name('prefix/input_image:0') ## x = input image
y = graph.get_tensor_by_name('prefix/output_7:0')  ## y = output

在此之后，我打开我的图像并对其进行整形

img = Image.open('asgdvd.jpg')

img = img.resize((1024,1024))
image = array(img).reshape(1,1024,3)

此后，我尝试使用以下代码片段查看预测。

with tf.Session(graph=graph) as sess:


    pred_y = sess.run(y,Feed_dict={x:image})
    print(pred_y)

我收到的是这样的： [[[-0.44022697 0.9068849 -0.57112163 3.5229325 ] [-0.9684447 0.8227438 -0.3559094 0.98766834] [-0.9200943 -0.7196723 2.4025247 1.1918476 ] ... [ 0.30505624 -0.2963381 0.51927084 4.303519 ] [ 0.3955547 0.4455924 2.9750154 4.038637 ] [-0.026547 -0.07817449 2.7281873 2.9988997 ]]]

形状 [1,261888,4]

我认为我拥有的是锚框，因为当我进行演示对象检测时，我得到了相同的锚框尺寸。

我正在尝试遵循这种方法： Import a simple Tensorflow frozen_model.pb file and make prediction in C++