From 5b70399f355fb93236632d5d3d32b8ac308094e5 Mon Sep 17 00:00:00 2001
From: Luca Marini <47863050+lucamarini22@users.noreply.github.com>
Date: Fri, 7 Apr 2023 09:20:55 +0200
Subject: [PATCH] Make AutoContrast work

---
 datasets/augmentation/ms_augmentations.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/datasets/augmentation/ms_augmentations.py b/datasets/augmentation/ms_augmentations.py
index 87c77c2..d121967 100644
--- a/datasets/augmentation/ms_augmentations.py
+++ b/datasets/augmentation/ms_augmentations.py
@@ -6,7 +6,7 @@
 
 
 def AutoContrast(img, v):
-    return iaaa.pillike.Autocontrast(cutoff=0)(image=img)["image"]
+    return iaaa.pillike.Autocontrast(cutoff=v)(image=img)
 
 
 def Brightness(img, v):
@@ -65,7 +65,7 @@ def TranslateY(img, v):
 def ms_augmentation_list():
     l = [
         # The below four don't work with multispectral images
-        # (AutoContrast, 0, 1),
+        (AutoContrast, 0, 20),
         # (Brightness, 0.05, 0.95),
         # (Color, 0.05, 0.95),
         # (Contrast, 0.05, 0.95),