上采样函数pixel_shuffle的使用

非科普，仅记录自己在使用中报错和解决问题情况，方便以后回顾。随缘更

pixel_shuffle = torch.nn.PixelShuffle(2) #括号中的数值只能是int不能是float,表示上采样倍率
output = pixel_shuffle(img_tensor)#变量必须是Tensor型

把图片转成Tensor:

tensor = transforms.ToTensor() # tensor数据格式是torch(C,H,W)
img_tensor=tensor(img)

opencv储存图片的格式和torch的储存方式不一样，opencv储存图片格式是（H,W,C），而torch储存的格式是（C,H,W).如果使用cv2.imread读入img，直接转换成Tensor就会报错，需要先对图片进行转换。这里有一个简单的方法就是直接使用plt.imread读入图片。

运行结果：