可视化语义分割结果
阴阳互根def test_visul_label():
path='./train_annot'
output='./train_annot_out'
if not ists(output):
os.mkdir(output)
label_path=[os.path.join(path,i) for i in os.listdir(path)]
for label_path_ in label_path:
label = cv2.imread(label_path_)
print(label.shape)
print(type(label))姜子牙怎么玩
label =label[:, :, 0]
形容疼痛的成语cmap = np.array([[0, 0, 0],
[128, 0, 0],
[128, 128, 0],测绘师
[0, 128, 0],
[0, 0, 128]]
)
y = label
r = y.copy()
g = y.copy()
b = y.copy()偶数和奇数是什么意思
婚礼祝福词肉咪咪# print('r=',r)
for l in range(0, len(cmap)):
r[y == l] = cmap[l, 0]
g[y == l] = cmap[l, 1]
高中历史必修三
b[y == l] = cmap[l, 2]
atenate((np.expand_dims(b,axis=-1),np.expand_dims(g,axis=-1),
cv2.imwrite(output+'/'+label_path_.split('/')[-1],label)
变为
