怎么用Tensorflow的预训练MaskRCNN实例分割模型？

Question

Tensorflow最近似乎发布了预训练模型，使用MaskRCNN进行实例分割，如下面的tweet所示。在

https://twitter.com/TensorFlow/status/963472849511092225

我下载了mask_rcnn_resnet101_atrus_coco_2018_01_28。焦油gz我从this中找到了冻结模型（pb）文件，并在Tensorboard中加载了该图。在

但是我不知道应该为模型输入什么，我找不到一个节点，在那里我可以简单地输入一个图像，虽然我能够找到节点，我们得到类，掩码，边界框等等

网上似乎也没有关于如何使用它的细节（可能是因为它是新的）

Answer 1

如果您遵循this tensorflow tutorial，它将向您展示如何在单个/组图像上运行冻结模型。要将此应用于下载的模型，最简单的方法是首先替换行：

PATH_TO_CKPT = MODEL_NAME + '/frozen_inference_graph.pb'

有一个下载模型的路径，即

PATH_TO_CKPT = '/absolute/path/to/frozen_inference_graph.pb'

那么就不需要运行Download Model部分下的代码。其他的也一样。在