如何使用Tensorflow的实例分割预训练的MaskRCNN模型？

Question

Tensorflow最近似乎已经发布了使用MaskRCNN进行实例分割的预训练模型，如下文所述。

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

我从这里下载了mask_rcnn_resnet101_atrous_coco_2018_01_28.tar.gz并试图弄清楚如何使用它。我找到了冻结的模型（pb）文件并将图形加载到Tensorboard中。

但是我不知道应该为模型输入什么。我找不到能够简单输入图像的节点，尽管我能够找到获得类，掩码，边界框等的节点。

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

Answer 1

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

PATH_TO_CKPT = MODEL_NAME + '/frozen_inference_graph.pb'

带有下载模型的路径，即

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

然后，无需运行“ Download Model ”部分下的代码。 其余的应该相同。