我的影像為 64x64 numpy 影像，如下所示：
此影像標記為 5。我有 5 個不同的類別（類別）。有什么辦法可以將此影像存盤為 RGB 或 (64,64,5)？
RG 作為中心影像，B 用標簽遮蓋。我有點困惑，我的主管對此非常含糊。或者它可能是 (64,64,2) 和第二個切片作為屏蔽標簽？
uj5u.com熱心網友回復：
您可以為每個類使用一種顏色，如下所示：
import cv2
# from google.colab.patches import cv2_imshow
import numpy as np
gray = cv2.imread('input.png', 0)
zero = np.zeros_like(gray)
# 5 classes, different color channels
rgb_1 = np.stack([gray, zero, zero], axis=2)
rgb_2 = np.stack([zero, gray, zero], axis=2)
rgb_3 = np.stack([gray, gray, zero], axis=2)
rgb_4 = np.stack([zero, zero, gray], axis=2)
rgb_5 = np.stack([gray, zero, gray], axis=2)
result = np.vstack([np.hstack([rgb_1, rgb_2, rgb_3]), np.hstack([rgb_4, rgb_5, rgb_5])])
# cv2_imshow(result)
cv2.imshow('result', result)
cv2.waitKey(0)
結果將是這樣的：
我只是為了可視化目的而復制最后一節課！
否則，如果您想在影像本身內部標記類，則可以使用Scikit-Image標記函式：from skimage.measure import label并from skimage.color import label2rgb按照此答案中所做的那樣。
轉載請註明出處，本文鏈接：https://www.uj5u.com/qianduan/535873.html
標籤：麻木的机器学习图像处理