support face enhance

2021-08-22 18:09:28 +08:00
parent 978def19a6
commit 667e34e7ca
5 changed files with 29 additions and 6 deletions
--- a/realesrgan/utils.py
+++ b/realesrgan/utils.py
@@ -143,7 +143,7 @@ class RealESRGANer():
        h_input, w_input = img.shape[0:2]
        # img: numpy
        img = img.astype(np.float32)
-        if np.max(img) > 255:  # 16-bit image
+        if np.max(img) > 256:  # 16-bit image
            max_range = 65535
            print('\tInput is a 16-bit image')
        else: