ValueError: cannot reshape array of size 435 into shape (145,145,3) about ssun HOT 3 OPEN

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

from ssun.

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.

from ssun.

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

from ssun.