使用scipy.io.loadmat()加载高光谱数据集.mat时报错:KeyError:‘indian_pines’处理办法分享

#最近在学习利用深度学习进行高光谱分类,下载了几个常用的高光谱公开数据集后,发现其是mat文件,可以用matlab打开,但在python中不能直接使用。

在python中加载mat文件需要使用scipy.io.loadmat()函数

参考前人的代码:

from scipy.io import loadmat
input_image = loadmat('D:\pycharmProj\HSIclassify\Pines\Indian_pines.mat')['indian_pines']
output_image = loadmat('D:\pycharmProj\HSIclassify\Pines\Indian_pines_gt.mat')['indian_pines_gt']

但运行之后程序报错:

 上CSDN查找解决方案,他提出在使用loadmat函数的地方Debug,并且在Debugger处敲入X.keys(),X就是你给loadmat函数读取的mat文件赋予的变量名,该博主演示如下:

但这种方法对我的代码似乎并不适用:

还有其他人说方括号里的字母都要小写,这显然是没道理的,因为我还是报错。


经过我的不断试错,终于解决了这个问题。因为loadmat函数读取出来的高光谱数据是dict格式的所以需要定位才能进行后续操作,定位通常是通过列名,所以找到正确的列名,并修改源代码,问题就能解决。可以直接在matlab中打开mat文件查看列名,或者利用python代码查看列名: 

input_image = loadmat('D:\pycharmProj\HSIclassify\Pines\Indian_pines.mat')
print(input_image.keys())
output_image = loadmat('D:\pycharmProj\HSIclassify\Pines\Indian_pines_gt.mat')
print(output_image.keys())

运行结果:

 可以看到列名是'indian_pines_corrected'和‘indian_pines_gt',所以正确的代码应该是:

再运行就不会报错啦!

写这个博客的目的是帮助像我一样的高光谱图像处理小白,大家共勉;也欢迎此邻域的大牛们莅临指导。

感谢您的阅读。

 

  • 6
    点赞
  • 13
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值