Sanatize label name for brushlabel conversion

Ensure to replace invalid characters of label names before converting to NumPy array or PNG images.
HumanSignal · Sep 4, 2023 · 10e33e5 · 10e33e5
1 parent 84369c3
commit 10e33e5
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/label_studio_converter/brush.py b/label_studio_converter/brush.py
@@ -146,6 +146,12 @@ def save_brush_images_from_annotation(
     )  # sanitize filename
 
     for name in layers:
+        # sanitize label name     
+        sanatized_name = name.replace('/', '-')
+        sanatized_name = sanatized_name.replace('\\', '-')
+        sanatized_name = sanatized_name.replace(' ', '-')
+        sanatized_name = sanatized_name.replace('.', '-')
+
         filename = os.path.join(
             out_dir,
             'task-'