沿z轴连接数据加载器

Question

我有一个9个波段的多光谱数据集。由于数据非常大，我将每个波段拆分为256 x 256个样本。所以我每个波段有16个这样的样本，我把它们保存在不同的文件夹里。现在我如何连接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)  

第二种方式：或火炬。猫

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

主要的区别是，当torch.stack沿着一个新的轴串联时，torch.cat只能在现有的轴上串联（这就是为什么需要reshape命令）。