pytorch的nn.Upsample,查看上采样效果
使用Image类读取图片,转为numpy数组,HWC转为CHW,然后转为[1,C,H,W],再转为Tensor,数据类型转为float。利用upsample进行上采样。再按相反的顺序,最终转为Image类的格式,显示出来。import torchimport torch.nn as nnfrom PIL import Imageimport numpy as npm = nn.Upsample(scale_factor=2, mode='nearest')up = nn.Upsample(
原创
2020-07-06 13:07:58 ·
5314 阅读 ·
0 评论