img是彩色圖像（3 個通道的數(shù)組）。img_sc一個與img大小相同但有 1 個通道的零數(shù)組。此操作將 rgb 圖像（包含 5 個類的分段圖像）的顏色映射到包含類 ID 的 ID 圖像：img_sc[np.logical_and(np.logical_and(img[:,:,0]==170,img[:,:,1]==170),img[:,:,2]==170)] = 1img_sc[np.logical_and(np.logical_and(img[:,:,0]==0,img[:,:,1]==255),img[:,:,2]==0)] = 2換句話說，在分割圖像中，道路是灰色的 (170, 170,170) 并且它的類 ID 是 1 ，草是 (0,255,0) 并且它的 ID 是 2 等到其他類，所以沒有圖像我有一個包含 1 個通道的數(shù)組，其中只包含 IDs 。我需要執(zhí)行與此操作相反的操作，并轉(zhuǎn)換包含 ID 的 1 通道數(shù)組，如下所示：[[3 3 3 ... 4 4 4] [3 3 3 ... 4 4 4] [3 3 3 ... 4 4 4] ... [3 3 3 ... 2 2 2] [3 3 3 ... 2 2 2] [3 3 3 ... 2 2 2]]進入包含 RGB 顏色的 3 通道數(shù)組：[[[102 102 51] [102 102 51] [102 102 51] ... [ 0 120 255] [ 0 120 255] [ 0 120 255]] [[102 102 51] [102 102 51] [102 102 51] ... [ 0 120 255] [ 0 120 255] [ 0 120 255]] [[102 102 51] [102 102 51] [102 102 51] ... [ 0 120 255] [ 0 120 255] [ 0 120 255]] ... [[102 102 51] [102 102 51] [102 102 51] ... [ 0 255 0] [ 0 255 0] [ 0 255 0]] [[102 102 51] [102 102 51] [102 102 51] ... [ 0 255 0] [ 0 255 0] [ 0 255 0]] [[102 102 51] [102 102 51] [102 102 51] ... [ 0 255 0] [ 0 255 0] [ 0 255 0]]]
2 回答
倚天杖
TA貢獻1828條經(jīng)驗 獲得超3個贊
h, w, c = img.shape
out = np.zeros((h,w,c))
out[class[:, :] == 1, :] = [170, 170, 170]
out[class[:, :] == 2, :] = [0, 255, 0]
可以工作，或者
h, w, c = img.shape
out = np.zeros((h,w,c))
out[class[:, :] == 1, 0] = 170
out[class[:, :] == 1, 1] = 170
out[class[:, :] == 1, 2] = 170
out[class[:, :] == 2, 1] = 255
陪伴而非守候
TA貢獻1757條經(jīng)驗 獲得超8個贊
arr = np.stack((arr,)*3, axis=-1)
arr[np.where((arr == 1).all(axis=2))] = [170, 170, 170]
添加回答
舉報
0/150
提交
取消