machine-learning - 在 tensorflow 目标检测中预测单张图像

Question

我使用 tensorflow object detection 训练了一个物体检测模型存储库。

现在正在评估default way是使用tf.contrib.slim.parallel_reader

从 record文件中读取

我想使用文件系统直接读取图像并获得图像的预测。我该怎么做？

Answer 1

看看这个 https://github.com/tensorflow/models/blob/master/object_detection/object_detection_tutorial.ipynb ，这里使用了卡住模型，但您可以使用正常的 ckpt 并稍微更改逻辑以使用图像而不是 tfrecord。

第1步:使用ckpt文件加载模型

step2:获取模型端点来运行预测；与上面的链接类似，只是删除了 :0

第三步:使用占位符输入 numpy 数组(您的图像)

除此之外，您还可以将训练后的模型转换为卡住模型 convert to frozen model并按照上述链接中所述的相同步骤进行操作。