在张量流物体检测中预测单个图像

时间:2017-08-03 07:09:04

标签: machine-learning tensorflow deep-learning object-detection predict

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

现在评估default way时,要使用recordtf.contrib.slim.parallel_reader文件中读取

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

1 个答案:

答案 0 :(得分:4)

看看这个https://github.com/tensorflow/models/blob/master/object_detection/object_detection_tutorial.ipynb,这里使用的是冻结模型,但你可以使用普通的ckpt和较小的改变逻辑来使用image而不是tfrecord。

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

第二步:获取模型终点以运行预测;类似于上面的链接,除了删除:0

第3步:使用占位符来提供numpy数组(您的图像)

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