OpenDroneMap-WebODM/app/api/custom_colormaps_helper.py

44 wiersze
833 B
Python

import math
from .common import hex2rgb
def discrete_cmap_from_hex(hex_colors):
rgb_colors = [hex2rgb(h, with_alpha=True) for h in hex_colors]
return {x: rgb_colors[math.floor(x / 256.0 * len(rgb_colors))] for x in range(0, 255)}
ndvi_arr = [
'#AD0028',
'#C5142A',
'#E02D2C',
'#EF4C3A',
'#FE6C4A',
'#FF8D5A',
'#FFAB69',
'#FFC67D',
'#FFE093',
'#FFEFAB',
'#FDFEC2',
'#EAF7AC',
'#D5EF94',
'#B9E383',
'#9BD873',
'#77CA6F',
'#53BD6B',
'#14AA60',
'#009755',
'#007E47'
]
contrast_ndvi_arr = [
'#AD0028',
'#FFAB69',
'#FDFEC2',
'#77CA6F',
'#014729'
]
custom_colormaps = [
{"discrete_ndvi": discrete_cmap_from_hex(contrast_ndvi_arr)},
{"better_discrete_ndvi": discrete_cmap_from_hex(ndvi_arr)},
#add custom maps here
]