在张量流对象检测中预测单个图像

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。

第一步：使用ckpt文件加载模型

step2：将模型终点指向运行预测；类似于上面的 link 除了删除 :0

第 3 步：使用占位符输入 numpy 数组（您的图像）

除此之外，您还可以将经过训练的模型转换为冻结模型convert to frozen model并按照上述link中描述的相同步骤进行操作。