高光谱图像深度学习后处理的准备工作

用pytorch进行高光谱图像处理时，dataload将数据传进tensor后会变成4维数据（个数，通道数，长，宽）
再保存结果的时候注意将图像转换为正常的（长，宽，通道数）的形式

1. 将gpu的数据先转换为cpu的数据

sr_img = sr_img.cpu()

2. 改变维数的表示方式（个数，通道数，长，宽）变成（个数，长，宽，通道数）的

sr_img = np.transpose(sr_img, (0, 2, 3, 1))

3. 删除数据的第一位（个数），即得到了最终输出的结果图像

sr_img = np.squeeze(sr_img, 0)

备注：普通图像同理
21世纪第21年21天的21点21分