Commit 5762c6d3 authored by moonbings's avatar moonbings
Browse files

Fix minor

parent 790eac16
...@@ -88,8 +88,8 @@ class SynthDoG(templates.Template): ...@@ -88,8 +88,8 @@ class SynthDoG(templates.Template):
roi = data["roi"] roi = data["roi"]
# split # split
split = self.split_indexes[idx % len(self.split_indexes)] split_idx = self.split_indexes[idx % len(self.split_indexes)]
output_dirpath = os.path.join(root, self.splits[split]) output_dirpath = os.path.join(root, self.splits[split_idx])
# save image # save image
image_filename = f"image_{idx}.jpg" image_filename = f"image_{idx}.jpg"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment