OpenCV-Applications/ColorMapCustomLegend.py at master · eeyribas/OpenCV-Applications · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
import cv2
import numpy as np
import matplotlib.pyplot as plt

def BuildLutImage(cmap, height):
    lut=BuildLut(cmap)
    image=np.repeat(lut[np.newaxis, ...], height, axis=0)
    return image

def BuildLut(cmap):
    lut=np.empty(shape=(256, 3), dtype=np.uint8)
    max=256
    lastVal, lastCol=cmap[0]
    for step, col in cmap[1:]:
        val=int(step*max)
        for i in range(3):
            lut[lastVal:val, i]=np.linspace(lastCol[i], col[i], val-lastVal)
        lastCol = col
        lastVal = val
    return lut

def ApplyColorMap1(gray, cmap):
    lut=BuildLut(cmap)
    s0, s1=gray.shape
    out=np.empty(shape=(s0, s1, 3), dtype=np.uint8)
    for i in range(3):
        out[..., i]=cv2.LUT(gray, lut[:, i])
    return out

def ApplyColorMap2(gray, cmap):
    lut1=BuildLut(cmap)
    lut2=np.reshape(lut1, (256, 1, 3))
    imgColor=cv2.applyColorMap(gray, lut2)
    return imgColor

def ShowWithMatplotlib(colorImg, title, pos):
    imgRGB=colorImg[:, :, ::-1]
    ax=plt.subplot(2, 2, pos)
    plt.imshow(imgRGB)
    plt.title(title)
    plt.axis('off')

grayImg=cv2.imread('images/lenna.png', cv2.IMREAD_GRAYSCALE)
plt.figure(figsize=(14, 6))
plt.suptitle("Custom color maps based on key colors and legend", fontsize=14, fontweight='bold')

custom1=ApplyColorMap1(grayImg, ((0, (255, 0, 255)), (0.25, (255, 0, 180)), (0.5, (255, 0, 120)),
                       (0.75, (255, 0, 60)), (1.0, (255, 0, 0))))
custom2=ApplyColorMap2(grayImg, ((0, (0, 255, 128)), (0.25, (128, 184, 64)), (0.5, (255, 128, 0)),
                       (0.75, (64, 128, 224)), (1.0, (0, 128, 255))))
legend1=BuildLutImage(((0, (255, 0, 255)), (0.25, (255, 0, 180)), (0.5, (255, 0, 120)),
                      (0.75, (255, 0, 60)), (1.0, (255, 0, 0))), 20)
legend2=BuildLutImage(((0, (0, 255, 128)), (0.25, (128, 184, 64)), (0.5, (255, 128, 0)),
                      (0.75, (64, 128, 224)), (1.0, (0, 128, 255))), 20)

ShowWithMatplotlib(legend1, "", 1)
ShowWithMatplotlib(custom1, "", 3)
ShowWithMatplotlib(legend2, "", 2)
ShowWithMatplotlib(custom2, "", 4)
plt.show()