[nodes] ImageSegmentation: use ChoiceParam instead of ListAttribute f…

…or validClasses
alicevision · Jul 10, 2023 · 5dbf109 · 5dbf109
1 parent 9d09e85
commit 5dbf109
Showing 1 changed file with 18 additions and 21 deletions.
diff --git a/meshroom/nodes/aliceVision/ImageSegmentation.py b/meshroom/nodes/aliceVision/ImageSegmentation.py
@@ -32,29 +32,26 @@ class ImageSegmentation(desc.AVCommandLineNode):
             uid=[0]
         ),
 
-        desc.ListAttribute(
-            elementDesc=desc.ChoiceParam(
-                name="className",
-                label="Class Name",
-                description="Class name to be added to the mask.",
-                value="person",
-                values=[
-                    "__background__",
-                    "aeroplane",
-                    "bicycle", "bird", "boat", "bottle", "bus",
-                    "car", "cat", "chair", "cow",
-                    "diningtable", "dog",
-                    "horse",
-                    "motorbike",
-                    "person", "pottedplant",
-                    "sheep", "sofa",
-                    "train", "tvmonitor"
-                ],
-                exclusive=True,
-                uid=[0]),
+        desc.ChoiceParam(
             name="validClasses",
             label="Classes",
-            description="Classes names which are to be considered."
+            description="Classes names which are to be considered.",
+            value=["person"],
+            values=[
+                "__background__",
+                "aeroplane",
+                "bicycle", "bird", "boat", "bottle", "bus",
+                "car", "cat", "chair", "cow",
+                "diningtable", "dog",
+                "horse",
+                "motorbike",
+                "person", "pottedplant",
+                "sheep", "sofa",
+                "train", "tvmonitor"
+            ],
+            exclusive=False,
+            uid=[0],
+            joinChar=",",
         ),
 
         desc.ChoiceParam(