3pp-mirror/Meshroom
1
0
Fork 0
mirror of https://github.com/alicevision/Meshroom.git synced 2025-06-03 19:31:58 +02:00
Code Issues Projects Releases Packages Wiki Activity

[nodes] ImageSegmentation: cosmetic changes

Browse source
This commit is contained in:
Loïc Vital 2023-06-21 15:29:18 +02:00
parent 8bd666207a
commit 84ce8f475e
1 changed files with 23 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -9,44 +9,44 @@ class ImageSegmentation(desc.AVCommandLineNode):
category = 'Utils' category = 'Utils'
documentation = ''' documentation = '''
Generate a mask with segmented labels for each pixel Generate a mask with segmented labels for each pixel.
''' '''
inputs = [ inputs = [
desc.File( desc.File(
name='input', name="input",
label='Input', label="Input",
description='SfMData file input', description="SfMData file input.",
value='', value="",
uid=[0], uid=[0],
), ),
desc.File( desc.File(
name="modelPath", name="modelPath",
label="Segmentation model path", label="Segmentation Model",
description="Weights file for the internal model", description="Weights file for the internal model.",
value="${ALICEVISION_SEMANTIC_SEGMENTATION_MODEL}", value="${ALICEVISION_SEMANTIC_SEGMENTATION_MODEL}",
uid=[0] uid=[0]
), ),
desc.ListAttribute( desc.ListAttribute(
elementDesc=desc.StringParam( elementDesc=desc.StringParam(
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="classname",
uid=[0]), uid=[0]),
name="validClasses", name="validClasses",
label="Valid classes", label="Classes",
description="Classes names which are to be considered" description="Classes names which are to be considered."
), ),
desc.ChoiceParam( desc.ChoiceParam(
name='verboseLevel', name="verboseLevel",
label='Verbose Level', label="Verbose Level",
description='verbosity level (fatal, error, warning, info, debug, trace).', description="Verbosity level (fatal, error, warning, info, debug, trace).",
value='info', value="info",
values=['fatal', 'error', 'warning', 'info', 'debug', 'trace'], values=["fatal", "error", "warning", "info", "debug", "trace"],
exclusive=True, exclusive=True,
uid=[], uid=[],
) )
@ -54,9 +54,9 @@ class ImageSegmentation(desc.AVCommandLineNode):
outputs = [ outputs = [
desc.File( desc.File(
name='output', name="output",
label='Masks Folder', label="Masks Folder",
description='Output path for the masks.', description="Output path for the masks.",
value=desc.Node.internalFolder, value=desc.Node.internalFolder,
uid=[], uid=[],
), ),