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

Question

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

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

我从 this 下载了 mask_rcnn_resnet101_atrous_coco_2018_01_28.tar.gz 并试图找出如何使用 it.I 找到的冻结模型 (pb) 文件并将图表加载到 Tensorboard 中。

但我无法弄清楚 model.I 的输入应该是什么 找不到可以简单地输入图像的节点，尽管我能够找到我们获得 [=25= 的节点], masks, bounding boxes 等

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

Answer 1

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

PATH_TO_CKPT = MODEL_NAME + '/frozen_inference_graph.pb'

带有您下载的模型的路径，即

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

那就不用运行段Download Model下的代码了。其余的应该是一样的。