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

这篇博客介绍了如何在PyTorch中处理高光谱图像,包括使用dataloaders将数据转化为4维张量,然后通过CPU进行维度调整和数据截取,以得到最终的输出图像。博主详细阐述了从GPU到CPU的数据迁移过程,并提供了关键的代码片段,如利用numpy的transpose和squeeze函数来改变维度顺序和移除不必要的维度。这对于理解深度学习中数据预处理和后处理的步骤至关重要。
摘要由CSDN通过智能技术生成

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

  1. 将gpu的数据先转换为cpu的数据
sr_img = sr_img.cpu()
  1. 改变维数的表示方式(个数,通道数,长,宽)变成(个数,长,宽,通道数)的
sr_img = np.transpose(sr_img, (0, 2, 3, 1))
  1. 删除数据的第一位(个数),即得到了最终输出的结果图像
sr_img = np.squeeze(sr_img, 0)

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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值