神经网络中图像的默认输入尺寸?

Question

在阅读一本PyTorch书籍时，我看到了这段代码，作者改变了轴的顺序。

img_t。permute(1,2,0)(将坐标轴的顺序从C × H × W改为H × W × C)

H x W x C是神经网络中输入图像的默认输入尺寸吗?

Answer 1

在PyTorch中，输入是形状为N x C x H x W的沐浴，因此N是批处理大小，C是图像通道的数量，H和W是高度和宽度，如您所知。但是当你使用cv2时，默认的图像形状是HxWxC，所以你需要为pytorch交换尺寸。