3pp-mirror/Meshroom
1
0
Fork 0
mirror of https://github.com/alicevision/Meshroom.git synced 2025-04-29 18:27:23 +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(
elementDesc=desc.StringParam(
elementDesc=desc.ChoiceParam(
name="className",
label="Class Name",
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]),
name="validClasses",
label="Classes",