如何将OpenCV框架推入张量流label_image

时间:2018-08-17 03:03:08

标签: python tensorflow transfer-learning

我正在尝试使用Tensorflow的label_image.py脚本从摄像机的视频流中推断图像帧。我正在使用opencv来获取numpy数组的图像帧。

我看到label_image.py使用函数read_tensor_from_image_file将文件加载到Tensor中。但是我已经在内存中将文件作为numpy数组...如何将numpy图像转换为张量t?一旦有了张量,我就可以将其输入以下代码并获得我的预测。

with tf.Session(graph=graph) as sess:
    results = sess.run(output_operation.outputs[0], {
      input_operation.outputs[0]: t
    })
results = np.squeeze(results)

0 个答案:

没有答案