kopia lustrzana https://github.com/OpenDroneMap/WebODM
85 wiersze
3.2 KiB
Python
85 wiersze
3.2 KiB
Python
import numpy as np
|
|
|
|
# Originally based on work by Frank and Evan
|
|
# Modified by Piero Toffanin for speed and lower memory usage
|
|
# The code here is re-licensed under AGPLv3, but is based on MIT
|
|
|
|
#******************************************************************************
|
|
# Copyright (c) 2009, Frank Warmerdam
|
|
# Copyright (c) 2010, Even Rouault <even dot rouault at mines-paris dot org>
|
|
#
|
|
# Permission is hereby granted, free of charge, to any person obtaining a
|
|
# copy of this software and associated documentation files (the "Software"),
|
|
# to deal in the Software without restriction, including without limitation
|
|
# the rights to use, copy, modify, merge, publish, distribute, sublicense,
|
|
# and/or sell copies of the Software, and to permit persons to whom the
|
|
# Software is furnished to do so, subject to the following conditions:
|
|
#
|
|
# The above copyright notice and this permission notice shall be included
|
|
# in all copies or substantial portions of the Software.
|
|
#
|
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
|
|
# OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
|
|
# THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
|
# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
|
|
# DEALINGS IN THE SOFTWARE.
|
|
#******************************************************************************
|
|
|
|
|
|
# =============================================================================
|
|
# rgb_to_hs()
|
|
#
|
|
# rgb comes in as [r,g,b] with values in the range [0,255]. The returned
|
|
# hsv values will be with hue and saturation in the range [0,1]
|
|
#
|
|
|
|
def rgb_to_hs( r, g, b ):
|
|
|
|
maxc = np.maximum(r,np.maximum(g,b))
|
|
minc = np.minimum(r,np.minimum(g,b))
|
|
|
|
# compute the difference, but reset zeros to ones to avoid divide by zeros later.
|
|
ones = np.ones((r.shape[0],r.shape[1]), dtype=np.uint8)
|
|
maxc_minus_minc = np.choose( minc==maxc, (maxc-minc,ones) )
|
|
|
|
s = np.divide((maxc-minc), np.maximum(ones,maxc), dtype=np.float32)
|
|
rc = np.divide((maxc-r), maxc_minus_minc, dtype=np.float32)
|
|
gc = np.divide((maxc-g), maxc_minus_minc, dtype=np.float32)
|
|
bc = np.divide((maxc-b), maxc_minus_minc, dtype=np.float32)
|
|
|
|
h = np.zeros((r.shape[0],r.shape[1]), dtype=np.float32)
|
|
np.choose( maxc==b, (h,4.0+gc-rc), out=h)
|
|
np.choose( maxc==g, (h,2.0+rc-bc), out=h)
|
|
np.choose( maxc==r, (h,bc-gc), out=h)
|
|
|
|
np.mod(h/6.0,1.0,out=h)
|
|
|
|
return h, s
|
|
|
|
# =============================================================================
|
|
# hsv_to_rgb()
|
|
#
|
|
# hsv comes in as [h,s,v] with hue and saturation in the range [0,1],
|
|
# but value in the range [0,255].
|
|
|
|
def hsv_to_rgb( h, s, v ):
|
|
i = (h*6.0).astype(int)
|
|
f = (h*6.0) - i
|
|
p = v*(1.0 - s)
|
|
q = v*(1.0 - s*f)
|
|
t = v*(1.0 - s*(1.0-f))
|
|
|
|
r = i.choose( v, q, p, p, t, v )
|
|
g = i.choose( t, v, v, q, p, p )
|
|
b = i.choose( p, p, t, v, v, q )
|
|
|
|
return np.asarray([r,g,b]).astype(np.uint8)
|
|
|
|
|
|
def hsv_blend(rgb, intensity):
|
|
h, s = rgb_to_hs(rgb[0], rgb[1], rgb[2])
|
|
#replace v with hillshade
|
|
#convert back to RGB
|
|
return hsv_to_rgb(h, s, intensity) |