search

mengduanjinghua / Spectral-spatial-Mamba-for-HSIC

Spectral-Spatial Mamba for Hyperspectral Image Classification, remote sensing
18 stars 3 forks source link

predicted_thematic_map[i, j, :] = indianpines_colors[test_pred[i, j]] #1

Open onceagain2001 opened 1 month ago

onceagain2001 commented 1 month ago

你好，在generate_pic.py中gt_thematic_map[i, j, :] = indianpines_colors[gt[i,j]] predicted_thematic_map[i, j, :] = indianpines_colors[gt[i,j]] 这里是不是应该改为predicted_thematic_map[i, j, :] = indianpines_colors[test_pred[i, j]]

mengduanjinghua commented 1 month ago

你好，在generate_pic.py中gt_thematic_map[i, j, :] = indianpines_colors[gt[i,j]] predicted_thematic_map[i, j, :] = indianpines_colors[gt[i,j]] 这里是不是应该改为predicted_thematic_map[i, j, :] = indianpines_colors[test_pred[i, j]] 这里仅仅是初始化一下，这部分后面的代码将模型对测试样本的预测结果赋给分类结果图。test_pred是按顺序展平后的测试样本的分类结果，在这里的test_pred不是二维数组。