如何使用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 ”部分下的代碼。 其余的應該相同。