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

Question

Tensorflow 提供了一个用于图像推理的 label_image.py 实现。这对磁盘上的图像非常有效。但我有一个案例，我正在从网络摄像头读取流媒体视频，我想对每个图像帧运行推理以检测摄像头馈送中的对象。

当前 label_image.py只接受磁盘上的图像并使用 read_tensor_from_image_file 将其转换为 Tensor。我如何从内存中的 Open CV 图像帧中获取具有必要预处理的 Tensor？

Answer 1

有3种方法。你可以

1. 将 numpy 数组馈送到 tensorflow，考虑 opencv python 包装器将图像读取为 numpy 数组，
2. tf.py_func 在您因为性能问题不想提供输入时很有用。
3. 另外，如果您使用 C++，您可以定义一个自定义操作来将 opencv mat 包装到一个张量中。