update

2023-11-01 14:46:51 +08:00
parent e45b89d111
commit e75c9a402a
1 changed files with 1 additions and 1 deletions
@@ -830,7 +830,7 @@ class PostProcess:

        height, width = source.shape[:2]
        small_source = cv2.resize(source, (self.img_size, self.img_size))
-        laplacian_diff = source.astype(np.float) - cv2.resize(small_source, (width, height)).astype(np.float)
+        laplacian_diff = source.astype(np.float64) - cv2.resize(small_source, (width, height)).astype(np.float64)
        result = (cv2.resize(result, (width, height)) + laplacian_diff).round().clip(0, 255).astype(np.uint8)
        if self.denoise:
            result = cv2.fastNlMeansDenoisingColored(result)