-
Notifications
You must be signed in to change notification settings - Fork 0
/
image_mask.py
34 lines (28 loc) · 1.02 KB
/
image_mask.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
from scipy.io import loadmat
from PIL import Image
import numpy as np
import matplotlib.pyplot as plt
from matplotlib import colors
color_mat = loadmat('E:/IEEE_TGRS_SSTlFormer-main/data/AVIRIS_colormap.mat')
color_mat_list = list(color_mat)
color_matrix = color_mat[color_mat_list[3]] #(17,3)
# pred = Image.open('barbara.tif')
# data_label = loadmat("E:/IEEE_TGRS_SSTlFormer-main/data/Barbara/barbara_gtChanges.mat")['HypeRvieW']
pred = Image.open('BayArea.tif')
data_label = loadmat("E:/IEEE_TGRS_SSTlFormer-main/data/BayArea/bayArea_gtChanges2.mat")['HypeRvieW']
# pred = Image.open('barbara.tif')
# data_label = loadmat("E:/IEEE_TGRS_SSTlFormer-main/data/BayArea/bayArea_gtChanges2.mat")['HypeRvieW']
data_label[data_label!=0]=1
pred = np.array(pred)
pred = pred/255
output=pred*data_label
output=output+data_label
# plt.subplot(1,1,1)
# plt.imshow(output, colors.ListedColormap(color_matrix))
# # plt.show()
# import ipdb; ipdb.set_trace()
im=Image.fromarray(output/2*255)
im=im.convert('RGB')
im.show()
import ipdb; ipdb.set_trace()
1