图片常用的结构:
eg1:
[batch_size , channels,hight, width]
eg2:
[batch_size , hight, width, channels]
该结构的存储方式:
属于多维数组的存储方式。
如何转换通道:
data = torch.tensor(img, dtype=torch.float32).permute(2, 0, 1)
图片常用的结构:
eg1:
[batch_size , channels,hight, width]
eg2:
[batch_size , hight, width, channels]
该结构的存储方式:
属于多维数组的存储方式。
如何转换通道:
data = torch.tensor(img, dtype=torch.float32).permute(2, 0, 1)