Open qixy13 opened 4 years ago
给预测的Mask上色，看下面代码 import cv2 import numpy as np import matplotlib import matplotlib.pyplot as plt from matplotlib import cm from matplotlib.colors import LinearSegmentedColormap
import cv2
import numpy as np
import matplotlib
import matplotlib.pyplot as plt
from matplotlib import cm
from matplotlib.colors import LinearSegmentedColormap
#在自建的colorabar ，得到自己想要的颜色 def get_spectral():
#在自建的colorabar ，得到自己想要的颜色
def get_spectral():
ALL = 0 VEGETATION = 1 ROAD = 4 BUILDING = 2 WATER = 3 colormap_float = np.zeros( (5, 3), np.float ) #新建一个数组用于存储颜色的数值
ALL = 0
VEGETATION = 1
ROAD = 4
BUILDING = 2
WATER = 3
colormap_float = np.zeros( (5, 3), np.float ) #新建一个数组用于存储颜色的数值
colormap_float[ALL, :] = [255, 255, 255] colormap_float[WATER, :] = [34, 180, 238] colormap_float[VEGETATION, :] = [159, 255, 84] colormap_float[BUILDING, :] = [255, 191, 0] # colormap_float[BUILDING, :] = [34, 180, 238] colormap_float[ROAD, :] = [38, 71, 139]
colormap_float[ALL, :] = [255, 255, 255]
colormap_float[WATER, :] = [34, 180, 238]
colormap_float[VEGETATION, :] = [159, 255, 84]
colormap_float[BUILDING, :] = [255, 191, 0]
# colormap_float[BUILDING, :] = [34, 180, 238]
colormap_float[ROAD, :] = [38, 71, 139]
return colormap_float
matplotlib.use("Qt5Agg") path = Satellite-Segmentation-master/segnet/pre1.png" p_mask = cv2.imread(path)
matplotlib.use("Qt5Agg")
path = Satellite-Segmentation-master/segnet/pre1.png"
p_mask = cv2.imread(path)
w, h, d = np.shape(p_mask) p_src = np.full((w, h, d), 0, dtype=np.uint8)
w, h, d = np.shape(p_mask)
p_src = np.full((w, h, d), 0, dtype=np.uint8)
colormap_float = get_spectral() rgb_table = LinearSegmentedColormap.from_list('sst cmap', colormap_float/255)
colormap_float = get_spectral()
rgb_table = LinearSegmentedColormap.from_list('sst cmap', colormap_float/255)
import cmaps plt.imshow((p_mask[..., 0]), cmap=rgb_table)
import cmaps
plt.imshow((p_mask[..., 0]), cmap=rgb_table)
给预测的Mask上色，看下面代码
import cv2import numpy as npimport matplotlibimport matplotlib.pyplot as pltfrom matplotlib import cmfrom matplotlib.colors import LinearSegmentedColormap#在自建的colorabar ，得到自己想要的颜色def get_spectral():ALL = 0VEGETATION = 1ROAD = 4BUILDING = 2WATER = 3colormap_float = np.zeros( (5, 3), np.float ) #新建一个数组用于存储颜色的数值colormap_float[ALL, :] = [255, 255, 255]colormap_float[WATER, :] = [34, 180, 238]colormap_float[VEGETATION, :] = [159, 255, 84]colormap_float[BUILDING, :] = [255, 191, 0]# colormap_float[BUILDING, :] = [34, 180, 238]colormap_float[ROAD, :] = [38, 71, 139]return colormap_floatmatplotlib.use("Qt5Agg")path = Satellite-Segmentation-master/segnet/pre1.png"p_mask = cv2.imread(path)w, h, d = np.shape(p_mask)p_src = np.full((w, h, d), 0, dtype=np.uint8)colormap_float = get_spectral()rgb_table = LinearSegmentedColormap.from_list('sst cmap', colormap_float/255)import cmapsplt.imshow((p_mask[..., 0]), cmap=rgb_table)