Add new parameters in ImageProcessing node

This commit is contained in:
Stella TAN 2022-08-10 11:46:24 +02:00 committed by Candice Bentéjac
parent b3c1196e41
commit 85a3f2afd2

View file

@ -287,6 +287,51 @@ Convert or apply filtering to the input images.
enabled=lambda node: node.noiseFilter.noiseEnabled.value,
),
]),
desc.GroupAttribute(name="nlmFilter", label="NL Means Denoising", description="NL Means Denoising Parameters.", joinChar=":", groupDesc=[
desc.BoolParam(
name='denoiseEnabled',
label='Enable',
description='Use Non-local Mean Denoising from OpenCV to denoise images',
value=False,
uid=[0],
),
desc.FloatParam(
name='filteringParamH',
label='H',
description='Parameter regulating filter strength. Bigger H value perfectly removes noise but also removes image details, smaller H value preserves details but also preserves some noise.',
value=5,
range=(1, 1000, 0.01),
uid=[0],
enabled=lambda node: node.nlmFilter.denoiseEnabled.value,
),
desc.FloatParam(
name='filteringParamHForColorComponents',
label='HColor',
description='Parameter regulating filter strength for color images only. Normally same as Filtering Parameter H. Not necessary for grayscale images',
value=5,
range=(0, 1000, 0.01),
uid=[0],
enabled=lambda node: node.nlmFilter.denoiseEnabled.value,
),
desc.IntParam(
name='templateWindowSize',
label='Template Window Size',
description='Size in pixels of the template patch that is used to compute weights. Should be odd.',
value=7,
range=(1, 101, 2),
uid=[0],
enabled=lambda node: node.nlmFilter.denoiseEnabled.value,
),
desc.IntParam(
name='searchWindowSize',
label='Search Window Size',
description='Size in pixels of the window that is used to compute weighted average for given pixel. Should be odd. Affect performance linearly: greater searchWindowsSize - greater denoising time.',
value=21,
range=(1, 1001, 2),
uid=[0],
enabled=lambda node: node.nlmFilter.denoiseEnabled.value,
),
]),
desc.ChoiceParam(
name='outputFormat',
label='Output Image Format',