python - TensorFlow：如何在 Python 中异步加载图像以进行流式视频预测

我正在训练一个模型，该模型将视频作为输入并进行每帧预测（具体而言，是对象跟踪）。

TensorFlow 提供了data.Dataset.prefetch()在训练期间异步加载图像的操作。这很有效，因为我的训练序列只有几帧长。但是，我还想在测试期间异步加载图像，其中序列可能长达数千帧。

在 TensorFlow 中是否有任何方便的方法来实现这一点？我想我可以将每个视频定义为一个Dataset对象？这看起来有点难看，因为我需要在图表中为我的数据集中的每个视频创建一个操作？

谢谢！

标签： pythontensorflowcomputer-vision