Fixed RandomSunFlare circles only using the first element of the src_…

…color. (#1285)
albumentations-team · Sep 16, 2022 · b1af92a · b1af92a
1 parent cffad9d
commit b1af92a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/albumentations/augmentations/transforms.py b/albumentations/augmentations/transforms.py
@@ -695,8 +695,8 @@ def get_params_dependent_on_targets(self, params):
             rad = random.randint(1, max(height // 100 - 2, 2))
 
             r_color = random.randint(max(self.src_color[0] - 50, 0), self.src_color[0])
-            g_color = random.randint(max(self.src_color[0] - 50, 0), self.src_color[0])
-            b_color = random.randint(max(self.src_color[0] - 50, 0), self.src_color[0])
+            g_color = random.randint(max(self.src_color[1] - 50, 0), self.src_color[1])
+            b_color = random.randint(max(self.src_color[2] - 50, 0), self.src_color[2])
 
             circles += [
                 (