使用torch.nn.functional.interpolate时候需要转换一下图片的shape
例如图片是(256,256,3,8),需要将其permute(2, 3, 0, 1)转成(3,8,256,256)才能用interpolate
代码:
images = torch.Tensor(images).permute(2, 3, 0, 1)
images = torch.nn.functional.interpolate(images,scale_factor=2, mode='nearest', align_corners=None)
images = images.permute(2, 3, 0, 1)
**不过我不知道换了之后会不会改变图片的像素点特征