我使用tensorflow object detection存储库训练了一个对象检测模型。
现在评估default way时,要使用record
从tf.contrib.slim.parallel_reader
文件中读取
我想直接使用文件系统读取图像并获取图像的预测。我该怎么做?
答案 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,并按照上述链接中所述的相同步骤进行操作。