AssertionError

[ArkZero35]

data_path = "/root/autodl-tmp/HSIMAE/HSIMAE/Finetuning dataset/Envi_standard_format/WHU-Hi-LongKou/WHU-Hi-LongKou" # numpy array, [h, w, Bands]
gt_path = "/root/autodl-tmp/HSIMAE/HSIMAE/Finetuning dataset/Envi_standard_format/WHU-Hi-LongKou/WHU-Hi-LongKou" # numpy array, [h, w]

When I try to use WHU-Hi-LongKou dataset, "assert len(data_cubes) == gt.shape[0]" can not pass.
Becasues data_cubes.shape is (220000, 9, 9, 32), but gt.shape is (59400000,)
Please tell me what should I do, thank you very much.

[Ryan21wy]

The size of WHU-Hi-LongKou dataset was 550×400×270 (h×w×Bands), and 550×400=220,000, 550×400×270=59,400,000. The shape of gt should be (550×400) and reshape to (220,000,). Maybe you treat the HSI data as gt?

[ArkZero35]

print(gt_raw.shape) #(550, 400, 270) But as my experiment show, the shape of gt is (550,400,270) too. The file path is :envi_gt_path : /root/autodl-tmp/HSIMAE/HSIMAE/Finetuning dataset/Envi_standard_format/WHU-Hi-LongKou/WHU-Hi-LongKou_gt.hdr 你好朋友，我在自己做实验检查gt的时候，路径如上所示，它的数组形状也是(550,400,270)，所以我不太懂

…

------------------ 原始邮件 ------------------ 发件人: ***@***.***>; 发送时间: 2024年12月2日(星期一) 下午3:28 收件人: ***@***.***>; 抄送: ***@***.***>; ***@***.***>; 主题: Re: [Ryan21wy/HSIMAE] AssertionError (Issue #3) The size of WHU-Hi-LongKou dataset was 550×400×270 (h×w×Bands), and 550×400=220,000, 550×400×270=59,400,000. The shape of gt should be (550×400) and reshape to (220,000,). Maybe you treat the HSI data as gt? — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: ***@***.***>

[Ryan21wy]

那可能是Envi格式的问题，可以试试Matlab格式，在这里下载 http://rsidea.whu.edu.cn/resource_WHUHi_sharing.htm