update 30
This commit is contained in:
parent
9d1395a2a0
commit
80ce638013
1 changed files with 10 additions and 0 deletions
10
lib/util.py
10
lib/util.py
|
@ -58,3 +58,13 @@ def monoize(s):
|
||||||
s = np.sum(s, 1)
|
s = np.sum(s, 1)
|
||||||
s /= channels
|
s /= channels
|
||||||
return s
|
return s
|
||||||
|
|
||||||
|
def div0(a, b):
|
||||||
|
"""division, whereby division by zero equals zero"""
|
||||||
|
# http://stackoverflow.com/a/35696047
|
||||||
|
a = np.asanyarray(a)
|
||||||
|
b = np.asanyarray(b)
|
||||||
|
with np.errstate(divide='ignore', invalid='ignore'):
|
||||||
|
c = np.true_divide(a, b)
|
||||||
|
c[~np.isfinite(c)] = 0 # -inf inf NaN
|
||||||
|
return c
|
||||||
|
|
Loading…
Reference in a new issue