added support for input images with float datatype

mapbox · May 25, 2020 · 05e82db · 05e82db
1 parent d8a3f3f
commit 05e82db
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/rio_hist/utils.py b/rio_hist/utils.py
@@ -26,7 +26,8 @@ def reshape_as_raster(arr):
 def cs_forward(arr, cs='rgb'):
     """ RGB (any dtype) to whatevs
     """
-    arrnorm_raw = arr.astype('float64') / np.iinfo(arr.dtype).max
+    info = np.finfo if 'f' in arr.dtype.str else np.iinfo
+    arrnorm_raw = arr.astype('float64') / info(arr.dtype).max
     arrnorm = arrnorm_raw[0:3]
     cs = cs.lower()
     if cs == 'rgb':