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

Question

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

systemLayoutSizeFittingSize:withHorizontalFittingPriority:verticalFittingPriority:

我从https://twitter.com/TensorFlow/status/963472849511092225下载了mask_rcnn_resnet101_atrous_coco_2018_01_28.tar.gz，并试图弄清楚如何使用它。我找到了冻结模型（pb）文件，并在Tensorboard中加载了图形。

但我无法弄清楚模型的输入应该是什么。我找不到一个只能输入图像的节点，尽管我能够找到我们得到类，掩码，边界框等的节点。

此外，似乎没有关于如何使用它的在线细节（可能是因为它是新的）

Answer 1

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

const Circle = ({pct, color}) => { const style = { background: `linear-gradient(transparent ${100-pct}%, ${color} ${100-pct}%)` }; return <div className="circle" style={style}></div> } ReactDOM.render( <Circle pct={25} color={'orange'}/>, document.getElementById("root") );

带有下载模型的路径，即

PATH_TO_CKPT = MODEL_NAME + '/frozen_inference_graph.pb'

然后，无需在PATH_TO_CKPT = '/absolute/path/to/frozen_inference_graph.pb'部分下运行代码。其余的应该是一样的。