Skip to content

Commit b7877b7

Browse files
jasonrock-a3Dipet
andauthored
Fix 1072 sunflare dump better (albumentations-team#1333)
* simple fix * generate additional centers on a line * formatting Co-authored-by: Mikhail Druzhinin <[email protected]>
1 parent 57988c2 commit b7877b7

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

albumentations/augmentations/transforms.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -685,10 +685,13 @@ def get_params_dependent_on_targets(self, params):
685685
x = []
686686
y = []
687687

688-
for rand_x in range(0, width, 10):
689-
rand_y = math.tan(angle) * (rand_x - flare_center_x) + flare_center_y
688+
def line(t):
689+
return (flare_center_x + t * math.cos(angle), flare_center_y + t * math.sin(angle))
690+
691+
for t_val in range(-flare_center_x, width - flare_center_x, 10):
692+
rand_x, rand_y = line(t_val)
690693
x.append(rand_x)
691-
y.append(2 * flare_center_y - rand_y)
694+
y.append(rand_y)
692695

693696
for _i in range(num_circles):
694697
alpha = random.uniform(0.05, 0.2)

0 commit comments

Comments
 (0)