可视化颜色映射

label_colours = np.random.randint(255, size=(100, 3))
im_target = target.data.cpu().numpy()
im_target_rgb = np.array([label_colours[c % 100] for c in im_target])
im_target_rgb = im_target_rgb.reshape(im.shape).astype(np.uint8)
cv2.imwrite("output_%s_128_%is.jpg" % (args.input[6:-4], time1), im_target_rgb)

高光谱图像提取波段显示伪彩图(rgb)
https://www.okcode.net/article/96689

你可能感兴趣的:(可视化颜色映射)