如何保存PyTorch DataLoader生成的图片？

Question

我正在尝试将变换添加到我拥有的视频数据集，并且我已经提取了帧。我想要添加变换，但不是将变量保存到Dataloader中，而是将所有变换作为图像导出到新目录，这样我就可以使用它通过Video Swin Transformer - https://github.com/SwinTransformer/Video-Swin-Transformer训练模型

如何保存图片？

Answer 1

我认为您可以简单地遍历应用了转换的dataloader，并在生成的批处理中存储或直接保存每一帧。

如下所示：

transformed_images = []
for batch in dataloader:
    for video in batch:
        for frame in video:
            transformed_images.append(image)

取决于序列维度在数据中的位置