3pp-mirror/Meshroom
1
0
Fork 0
mirror of https://github.com/alicevision/Meshroom.git synced 2025-04-30 02:37:26 +02:00
Code Issues Projects Releases Packages Wiki Activity

[nodes] ImageSegmentation: ChoiceParam instead of StringParam for classe names

Browse source
This commit is contained in:
Loïc Vital 2023-06-21 15:57:14 +02:00
parent 7a20d27365
commit d4d5169d15
1 changed files with 15 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
meshroom/nodes/aliceVision/ImageSegmentation.py
View file

@ -30,11 +30,24 @@ Generate a mask with segmented labels for each pixel.
), ),
desc.ListAttribute( desc.ListAttribute(
elementDesc=desc.StringParam( elementDesc=desc.ChoiceParam(
name="className", name="className",
label="Class Name", label="Class Name",
description="Class name to be added to the mask.", description="Class name to be added to the mask.",
value="classname", 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]), uid=[0]),
name="validClasses", name="validClasses",
label="Classes", label="Classes",