Merge pull request #465 from drozbay/dev-pr

fix Image Levels Adjustment function to use mid_level
2024-08-24 19:52:12 -07:00
parent ee2e31a1e5 3f345aa190
commit fdb0f52e31
1 changed files with 7 additions and 4 deletions
@@ -6212,16 +6212,19 @@ class WAS_Image_Levels:

        def adjust(self, im):

-            im_arr = np.array(im)
+            im_arr = np.array(im).astype(np.float32)
            im_arr[im_arr < self.min_level] = self.min_level
            im_arr = (im_arr - self.min_level) * \
                (255 / (self.max_level - self.min_level))
-            im_arr[im_arr < 0] = 0
-            im_arr[im_arr > 255] = 255
+            im_arr = np.clip(im_arr, 0, 255)
+
+            # mid-level adjustment
+            gamma = math.log(0.5) / math.log((self.mid_level - self.min_level) / (self.max_level - self.min_level))
+            im_arr = np.power(im_arr / 255, gamma) * 255
+
            im_arr = im_arr.astype(np.uint8)

            im = Image.fromarray(im_arr)
-            im = ImageOps.autocontrast(im, cutoff=self.max_level)

            return im