Recreate colormap based on colorscale in matplotlib-CodePudding

I have an image of a color scale

I filter out the actual color scale by using

import cv2
import numpy as np

colorbar = cv2.imread('colorbar-scheme-elevation.png', cv2.IMREAD_UNCHANGED)
colorbar = cv2.cvtColor(colorbar, cv2.COLOR_BGRA2BGR)

hsv = cv2.cvtColor(colorbar, cv2.COLOR_RGB2HSV)
lower_gray = np.array([0, 0, 0])
upper_gray = np.array([255, 10, 255])

mask = cv2.inRange(hsv, lower_gray, upper_gray)
mask = cv2.bitwise_not(mask)
res = cv2.bitwise_and(colorbar, colorbar, mask=mask)
gray = cv2.cvtColor(res, cv2.COLOR_RGB2GRAY)


ret, thresh = cv2.threshold(gray, 10, 255, 0)
contours, hierarchy = cv2.findContours(thresh, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_NONE)
c = max(contours, key=cv2.contourArea)
x, y, w, h = cv2.boundingRect(c)

ROI = colorbar[y:y   h, x:x   w]

The end result looks as follows:

Next, I only need one row

ROI = ROI[0]

All the unique colors can be found using

unique_colors = list(reversed(np.unique(ROI, axis=0)))

How can I recreate the color scale with the unique color-values?

CodePudding user response：

You can create a Matplotlib colormap starting from a list of colors with ListedColormap (note that I didn't consider your last two commands):

from matplotlib.colors import ListedColormap
cmap = ListedColormap(ROI[0].astype(float) / 255)

CodePudding user response：

You can create a single row of ROI:

single_row_ROI = ROI[0:1,:]

All the unique colors can be put into a list:

colormap_list = single_row_ROI[0].tolist()

List containing first 10 colors:

>>> colormap_list[:10]
[[238, 135, 123], [237, 135, 123], [238, 136, 123], [238, 135, 123], [238, 135, 123], [237, 136, 123], [237, 135, 123], [237, 135, 123], [237, 136, 123], [237, 136, 123]]