基于Tensorflow的流计算

Question

我正在尝试使用Tensorflow在一个大的视频数据量(大约100 of )上运行一个计算。计算图将是一个简单的管道，由两个部分组成，一个运行在CPU上，另一个运行在GPU上。管道运行在单独的框架上，我不需要做任何训练。

因此，假设其他线程负责将帧加载到FIFOQueue queue_in并从queue_out中卸载它们，代码如下所示：

frame = queue_in.dequeue()
output = queue_out.enqueue(gpu_op(cpu_op(frame)))

with tf.Session():
    output.run()

我的问题是:我是否可以运行这个管道，以便GPU op和CPU op在两个连续的帧上同时运行？

谢谢!

Answer 1

我建议在工作中使用tf.data。下面的性能指南讨论了如何实现您感兴趣的流水线。