mirror of
https://github.com/alicevision/Meshroom.git
synced 2025-04-30 10:47:34 +02:00
132 lines
4 KiB
Python
132 lines
4 KiB
Python
from meshroom.core import desc
|
|
|
|
class DepthMap(desc.CommandLineNode):
|
|
internalFolder = '{cache}/{nodeType}/{uid0}/'
|
|
commandLine = 'aliceVision_depthMapEstimation {allParams}'
|
|
gpu = desc.Level.INTENSIVE
|
|
size = desc.DynamicNodeSize('ini')
|
|
parallelization = desc.Parallelization(blockSize=3)
|
|
commandLineRange = '--rangeStart {rangeStart} --rangeSize {rangeBlockSize}'
|
|
|
|
inputs = [
|
|
desc.File(
|
|
name="ini",
|
|
label='MVS Configuration file',
|
|
description='',
|
|
value='',
|
|
uid=[0],
|
|
),
|
|
desc.IntParam(
|
|
name='sgmMaxTCams',
|
|
label='SGM Max TCams',
|
|
description='Semi Global Matching Number of neighbour cameras.',
|
|
value=10,
|
|
range=(1, 100, 1),
|
|
uid=[0],
|
|
),
|
|
desc.IntParam(
|
|
name='sgmWSH',
|
|
label='SGM WSH',
|
|
description='Semi Global Matching Size of the patch used to compute the similarity.',
|
|
value=4,
|
|
range=(1, 20, 1),
|
|
uid=[0],
|
|
),
|
|
desc.FloatParam(
|
|
name='sgmGammaC',
|
|
label='SGM GammaC',
|
|
description='Semi Global Matching GammaC threshold.',
|
|
value=5.5,
|
|
range=(0.0, 30.0, 0.5),
|
|
uid=[0],
|
|
),
|
|
desc.FloatParam(
|
|
name='sgmGammaP',
|
|
label='SGM GammaP',
|
|
description='Semi Global Matching GammaP threshold.',
|
|
value=8.0,
|
|
range=(0.0, 30.0, 0.5),
|
|
uid=[0],
|
|
),
|
|
desc.IntParam(
|
|
name='refineNSamplesHalf',
|
|
label='Refine Number of samples',
|
|
description='Refine Number of samples.',
|
|
value=150,
|
|
range=(1, 500, 10),
|
|
uid=[0],
|
|
),
|
|
desc.IntParam(
|
|
name='refineNDepthsToRefine',
|
|
label='Refine Number of depths',
|
|
description='Refine Number of depths.',
|
|
value=31,
|
|
range=(1, 100, 1),
|
|
uid=[0],
|
|
),
|
|
desc.IntParam(
|
|
name='refineNiters',
|
|
label='Refine Number of iterations',
|
|
description='Refine: Number of iterations.',
|
|
value=100,
|
|
range=(1, 500, 10),
|
|
uid=[0],
|
|
),
|
|
desc.IntParam(
|
|
name='refineWSH',
|
|
label='Refine WSH',
|
|
description='Refine Size of the patch used to compute the similarity.',
|
|
value=3,
|
|
range=(1, 20, 1),
|
|
uid=[0],
|
|
),
|
|
desc.IntParam(
|
|
name='refineMaxTCams',
|
|
label='Refine Max TCams',
|
|
description='Refine Number of neighbour cameras.',
|
|
value=6,
|
|
range=(1, 20, 1),
|
|
uid=[0],
|
|
),
|
|
desc.FloatParam(
|
|
name='refineSigma',
|
|
label='Refine Sigma',
|
|
description='Refine Sigma threshold.',
|
|
value=15,
|
|
range=(0.0, 30.0, 0.5),
|
|
uid=[0],
|
|
),
|
|
desc.FloatParam(
|
|
name='refineGammaC',
|
|
label='Refine GammaC',
|
|
description='Refine GammaC threshold.',
|
|
value=15.5,
|
|
range=(0.0, 30.0, 0.5),
|
|
uid=[0],
|
|
),
|
|
desc.FloatParam(
|
|
name='refineGammaP',
|
|
label='Refine GammaP',
|
|
description='Refine GammaP threshold.',
|
|
value=8.0,
|
|
range=(0.0, 30.0, 0.5),
|
|
uid=[0],
|
|
),
|
|
desc.BoolParam(
|
|
name='refineUseTcOrRcPixSize',
|
|
label='Refine Use Tc or Rc pixel size',
|
|
description='Refine Use current camera pixel size or minimum pixel size of neighbour cameras.',
|
|
value=False,
|
|
uid=[0],
|
|
),
|
|
]
|
|
|
|
outputs = [
|
|
desc.File(
|
|
name='output',
|
|
label='Output',
|
|
description='Output folder for generated depth maps.',
|
|
value='{cache}/{nodeType}/{uid0}/',
|
|
uid=[],
|
|
),
|
|
]
|