Tensorflow如何从OpenCV图像帧读取张量

Question

Tensorflow提供了label_image.py实现来推断图像。这非常适合磁盘上的图像。但是我有一种情况，我正在从网络摄像头读取流视频，并且我想对每个图像帧进行推理以检测照相机源中的物体。

当前label_image.py仅接受磁盘上的图像，使用read_tensor_from_image_file会将其转换为张量。如何从内存中的Open CV图像帧中获得在read_tensor_from_image_file中进行了必要的预处理的Tensor？

Answer 1

有3种方法。你可以

1. 将numpy数组供入tensorflow，考虑将opencv python包装器读取的图像作为numpy数组，
2. tf.py_func在由于性能问题而不想输入您的输入时很有用。
3. 此外，如果您使用c ++，则可以定义一个自定义op来将opencv mat包装到张量。