pytorch - 用于 pytorch 中 RNN 的 Nvidia DALI

Question

我已经阅读了一些 Nvidia DALI Dataloader 代码。似乎所有这些都是为 CNN 实现的。有什么方法可以为 RNN 编写 DALI Dataloader？我正在使用 fn.external_source 来读取图像。

对于 CNN 案例，我只使用 f = open(img_path, 'rb'), image.append(np.frombuffer(f.read(), dtype=self.dtype)) 来创建一批所有图像的列表. 然后它将被解码，裁剪，在管道中调整大小。它确实有效。

现在我想对 RNN 做同样的事情。一批输入应该是 (batch_size, time_sequence, c,h,w)。我使用包含图像序列的列表列表，表示 [b1,b2,...b16]，每个 b_i 是包含列表 [t1,t2,...t5] 的列表，其中 t_i 表示图像。但它没有用。fn.decoders.image 无法处理此问题。

任何人都可以帮助使用 RNN 实现的数据加载器吗？

谢谢

标签： pytorchnvidiarecurrent-neural-networkdataloader