JPEGCompression bugfix

Fixed bug with quality range in JPEGCompression transform
imedslab · Mar 3, 2020 · e25bf0a · e25bf0a
1 parent 9ca08af
commit e25bf0a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/solt/transforms/_transforms.py b/solt/transforms/_transforms.py
@@ -1426,7 +1426,7 @@ def __init__(self, p=0.5, quality_range=None, data_indices=None):
         self.quality_range = validate_numeric_range_parameter(quality_range, (100, 100), 0, 100)
 
     def sample_transform(self, data):
-        self.state_dict["quality"] = random.randint(self.quality_range[0], self.quality_range[0])
+        self.state_dict["quality"] = random.randint(self.quality_range[0], self.quality_range[1])
 
     @img_shape_checker
     def _apply_img(self, img: np.ndarray, settings: dict):