fix: fix #69

opendatalab · Nov 15, 2024 · 0baa13c · 0baa13c
1 parent f4731bc
commit 0baa13c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/labelu/internal/common/converter.py b/labelu/internal/common/converter.py
@@ -348,8 +348,8 @@ def convert_to_mask(
             file_relative_path_model_l = f"{file_relative_path_base_name}-trainIds.png"
             file_full_path_model_l = out_data_dir.joinpath(file_relative_path_model_l)
             img_model_l = Image.new("L", (width, height), 0)
-            for p in polygons:
-                ImageDraw.Draw(img_model_l).polygon(p, outline=1, fill=1)
+            for _index, p in enumerate(polygons):
+                ImageDraw.Draw(img_model_l).polygon(p, outline=1, fill=_index + 1)
             img_model_l.save(file_full_path_model_l, "PNG")
             export_files.append(file_full_path_model_l)