ValueError: cannot reshape array of size 435 into shape (145,145,3)

[sandeep-25]

labels=sio.loadmat("Indian_pines_gt.mat")['indian_pines_gt']
palette=np.array([[255,0,0],[0,255,0],[0,0,255],[255,255,0],[0,255,255],[255,0,255],[176,48,96],[46,139,87],[160,32,240],[255,127,80],[127,255,212],[218,112,214],[160,82,45],[127,255,0],[216,191,216],[238,0,0]])
palette=palette*1.0/255
X_result=np.zeros((labels.shape[0],3))
num_class=labels.max()
or i in range(0,num_class):
X_result[np.where(labels==i),0]=palette[i,0]
X_result[np.where(labels==i),1]=palette[i,1]
X_result[np.where(labels==i),2]=palette[i,2]
X_result=np.reshape(X_result,(145,145,3))

I am getting the following error,
ValueError: cannot reshape array of size 435 into shape (145,145,3)

[YonghaoXu]

Hi, I think the labels here should be reshaped into a vector form (e.g., labels = labels.reshape(145*145, 1)).

[sandeep-25]

X_result=np.zeros((labels.shape[0]*labels.shape[1],3))
num_class=labels.max()
X_result = np.zeros((labels.shape[0]*labels.shape[1],3))
for i in range(0,num_class):
X_result[np.where(labels==i),0] = palette[i,0]
X_result[np.where(labels==i),1] = palette[i,1]
X_result[np.where(labels==i),2] = palette[i,2]
I repeated the experiment as per your comment but I am not getting the plot of the image.

[YonghaoXu]

Hi, you can use the matplotlib package to save the classification map.
import matplotlib.pyplot as plt
...
plt.imsave('clsmap.png',X_result)