Fix dtype of super resolved image

awesome-ml · Oct 27, 2020 · bbd15b0 · bbd15b0
1 parent d7fb8b2
commit bbd15b0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/infer.py b/infer.py
@@ -37,7 +37,7 @@ def main():
         sr = model.predict(np.expand_dims(low_res, axis=0))[0]
 
         # Rescale values in range 0-255
-        sr = ((sr + 1) / 2.) * 255
+        sr = (((sr + 1) / 2.) * 255).astype(np.uint8)
 
         # Convert back to BGR for opencv
         sr = cv2.cvtColor(sr, cv2.COLOR_RGB2BGR)