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

[nodes] DepthMap: Add customPatchPattern group parameter

Browse source
This commit is contained in:
Grégoire De Lillo 2023-04-13 18:28:32 +02:00 • committed by Candice Bentéjac
parent 2d552ef55d
commit 1e4672510e
1 changed files with 89 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

89
meshroom/nodes/aliceVision/DepthMap.py
View file

@ -419,6 +419,95 @@ Use a downscale factor of one (full-resolution) only if the quality of the input
enabled= lambda node: node.colorOptimization.colorOptimizationEnabled.value,
),
]),
desc.GroupAttribute(
name='customPatchPattern',
label='Custom Patch Pattern',
description='User custom patch pattern for similarity comparison.',
advanced=True,
group=None,
groupDesc=[
desc.BoolParam(
name='sgmUseCustomPatchPattern',
label='Enable for SGM',
description='Enable custom patch pattern for similarity volume computation at the SGM step .',
value=False,
uid=[0],
advanced=True,
),
desc.BoolParam(
name='refineUseCustomPatchPattern',
label='Enable for Refine',
description='Enable custom patch pattern for similarity volume computation at the Refine step .',
value=False,
uid=[0],
advanced=True,
),
desc.ListAttribute(
name="customPatchPatternSubparts",
label="Subparts",
description='User custom patch pattern subparts for similarity volume computation.',
advanced=True,
enabled= lambda node: (node.customPatchPattern.sgmUseCustomPatchPattern.value or node.customPatchPattern.refineUseCustomPatchPattern.value),
elementDesc=desc.GroupAttribute(
name='customPatchPatternSubpart',
label='Patch Pattern Subpart',
description='',
joinChar=":",
group=None,
groupDesc=[
desc.ChoiceParam(
name='customPatchPatternSubpartType',
label='Type',
description='Patch pattern subpart type.',
value='full',
values=['full', 'circle'],
exclusive=True,
uid=[0],
),
desc.FloatParam(
name='customPatchPatternSubpartRadius',
label='Radius / WSH',
description='Patch pattern subpart half-width or circle radius.',
value=2.5,
range=(0.5, 30.0, 0.1),
uid=[0],
),
desc.IntParam(
name='customPatchPatternSubpartNbCoords',
label='Coordinates',
description='Patch pattern subpart number of coordinates (for circle or ignore).',
value=12,
range=(3, 24, 1),
uid=[0],
),
desc.IntParam(
name='customPatchPatternSubpartLevel',
label='Level',
description='Patch pattern subpart image level.',
value=0,
range=(0, 2, 1),
uid=[0],
),
desc.FloatParam(
name='customPatchPatternSubpartWeight',
label='Weight',
description='Patch pattern subpart weight.',
value=1.0,
range=(0.0, 1.0, 0.1),
uid=[0],
),
]),
),
desc.BoolParam(
name='customPatchPatternGroupSubpartsPerLevel',
label='Group Subparts Per Level',
description='Group all subparts with the same image level.',
value=False,
uid=[0],
advanced=True,
enabled= lambda node: (node.customPatchPattern.sgmUseCustomPatchPattern.value or node.customPatchPattern.refineUseCustomPatchPattern.value),
),
]),
desc.GroupAttribute(
name='intermediateResults',
label='Intermediate Results',