Fix minor

5762c6d3 · moonbings · 790eac16 · 5762c6d3
Commit 5762c6d3 authored Nov 20, 2022 by moonbings
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

synthdog/template.py synthdog/template.py +2 -2

No files found.
--- a/synthdog/template.py
+++ b/synthdog/template.py
@@ -88,8 +88,8 @@ class SynthDoG(templates.Template):
        roi = data["roi"]

        # split
-        split = self.split_indexes[idx % len(self.split_indexes)]
-        output_dirpath = os.path.join(root, self.splits[split])
+        split_idx = self.split_indexes[idx % len(self.split_indexes)]
+        output_dirpath = os.path.join(root, self.splits[split_idx])

        # save image
        image_filename = f"image_{idx}.jpg"