看一下这个https://github.com/tensorflow/models/blob/master/object_detection/object_detection_tutorial.ipynb，这里使用了一个冻结的模型，但您可以使用普通的检查点，并稍微更改逻辑，以使用图像而不是tfrecord。

step1:使用ckpt文件加载模型

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

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

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

票数 4

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/45477199

复制

相似问题

问tensorflow目标检测中的单幅图像预测
EN

问tensorflow目标检测中的单幅图像预测EN