非科普,仅记录自己在使用中报错和解决问题情况,方便以后回顾。随缘更
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读入图片。
运行结果: