用pytorch进行高光谱图像处理时,dataload将数据传进tensor后会变成4维数据(个数,通道数,长,宽)
再保存结果的时候注意将图像转换为正常的(长,宽,通道数)的形式
- 将gpu的数据先转换为cpu的数据
sr_img = sr_img.cpu()
- 改变维数的表示方式(个数,通道数,长,宽)变成(个数,长,宽,通道数)的
sr_img = np.transpose(sr_img, (0, 2, 3, 1))
- 删除数据的第一位(个数),即得到了最终输出的结果图像
sr_img = np.squeeze(sr_img, 0)
备注:普通图像同理
21世纪第21年21天的21点21分