您好,欢迎来到化拓教育网。
搜索
您的当前位置:首页上采样函数pixel_shuffle的使用

上采样函数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读入图片。

运行结果:

因篇幅问题不能全部显示,请点此查看更多更全内容

Copyright © 2019- huatuo9.cn 版权所有 赣ICP备2023008801号-1

违法及侵权请联系:TEL:199 18 7713 E-MAIL:2724546146@qq.com

本站由北京市万商天勤律师事务所王兴未律师提供法律服务