沿 z 轴连接 PyTorch Dataloader

Question

我有一个包含 9 个波段的多光谱数据集。由于数据非常大，我将每个波段分成 256 x 256 个样本。所以我为每个乐队准备了 16 个这样的样本，并将它们保存到不同的文件夹中。现在如何连接 9 个波段的每个样本？

例如，我想将第一个波段数据中的第一个样本与第二个波段的第一个样本、第三个波段的第一个样本等连接起来，直到第 9 个波段。然后是第 1、第 2 ... 第 9 波段的第 2 个样本。以此类推，直到第 16 个样本。

Answer 1

您可以使用 (torch.stack)[https://pytorch.org/docs/stable/generated/torch.stack.html] 或 (torch.cat)[https://pytorch.org/docs /stable/generated/torch.cat.html]

例如，让我们生成一些随机的 256x256 矩阵：

import torch

a = torch.FloatTensor(256,256).uniform_(0,1)
b = torch.FloatTensor(256,256).uniform_(0,1)

第一种方式：您可以将张量与 torch.stack 连接。

c = torch.stack((a,b),axis=2)  

第二种方式：或torch.cat

c = torch.cat((a.reshape(256,256,-1),b.reshape(256,256,-1)),axis=2)

主要区别在于，torch.stack 沿新轴连接，而 torch.cat 只能在现有轴上连接（这就是需要 reshape 命令的原因）。