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

【问题标题】：How to use Instance segmentation pretrained MaskRCNN model by Tensorflow?如何使用 Tensorflow 的实例分割预训练 MaskRCNN 模型？
【发布时间】：2018-07-28 21:58:51
【问题描述】：

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

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

我从this 下载了 mask_rcnn_resnet101_atrous_coco_2018_01_28.tar.gz 并试图弄清楚如何使用它。我找到了冻结模型 (pb) 文件并在 Tensorboard 中加载了图形。

但我无法弄清楚模型的输入应该是什么。我找不到可以简单地输入图像的节点，尽管我能够找到我们获得类、掩码、边界框等的节点。

另外网上好像也没有具体的使用方法（可能是新的）

【问题讨论】：

你检查过更新的 .ipynb 教程吗？ github.com/tensorflow/models/blob/master/research/… 。它涵盖了在图像上运行模型的步骤。
感谢链接，它似乎有我正在寻找的东西。但是这个例子似乎是针对 mobilenet 架构的，但我希望 Resnet 也一样。

标签： python tensorflow mask

【解决方案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部分下的代码了。其余的应该是一样的。

【讨论】：