[英]map grayscale values to RGB values in image
讓我們考慮一個灰度值,其值在[0,255]范圍內。 我們如何有效地將每個值映射到RGB值?
到目前為止,我已經提出了以下實現:
# function for colorizing a label image:
def label_img_to_color(img):
label_to_color = {
0: [128, 64,128],
1: [244, 35,232],
2: [ 70, 70, 70],
3: [102,102,156],
4: [190,153,153],
5: [153,153,153],
6: [250,170, 30],
7: [220,220, 0],
8: [107,142, 35],
9: [152,251,152],
10: [ 70,130,180],
11: [220, 20, 60],
12: [255, 0, 0],
13: [ 0, 0,142],
14: [ 0, 0, 70],
15: [ 0, 60,100],
16: [ 0, 80,100],
17: [ 0, 0,230],
18: [119, 11, 32],
19: [81, 0, 81]
}
img_height, img_width = img.shape
img_color = np.zeros((img_height, img_width, 3))
for row in range(img_height):
for col in range(img_width):
label = img[row, col]
img_color[row, col] = np.array(label_to_color[label])
return img_color
但是,如您所見,它效率不高,因為有兩個“ for”循環。
在“ 將灰度值轉換為RGB表示形式”中也提出了這個問題? ,但未提出有效的實施建議。
一個更有效的方法是,而不是在所有像素上使用double for循環:
rgb_img = np.zeros((*img.shape, 3))
for key in label_to_color.keys():
rgb_img[img == key] = label_to_color[key]
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.