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

[nodes] KeyframeSelection: Add smart keyframe selection method

Browse source 
Group the keyframe selection parameters together depending on the method
(regular or smart) they belong to.

Parameters for a method are only shown if that method is the currently
selected one.
This commit is contained in:
Candice Bentéjac 2023-01-24 15:16:42 +00:00
parent e09554c410
commit 30ea556c26
1 changed files with 113 additions and 37 deletions
Download patch file Download diff file Expand all files Collapse all files

150
meshroom/nodes/aliceVision/KeyframeSelection.py
View file

@ -73,49 +73,125 @@ You can extract frames at regular interval by configuring only the min/maxFrameS
uid=[0], uid=[0],
), ),
desc.GroupAttribute( desc.GroupAttribute(
name="regularSelection", name="selectionMethod",
label="Regular Keyframe Selection", label="Keyframe Selection Method",
description="Parameters for the regular keyframe selection.\nKeyframes are selected regularly over the sequence with respect to the set parameters.", description="Selection the regular or smart method for the keyframe selection.\n"
"- With the regular method, keyframes are selected regularly over the sequence with respect to the set parameters.\n"
"- With the smart method, keyframes are selected based on their sharpness and optical flow scores.",
group=None, # skip group from command line group=None, # skip group from command line
groupDesc=[ groupDesc=[
desc.BoolParam( desc.BoolParam(
name="useRegularSelection", name='useSmartSelection',
label="Use Regular Selection", label='Use Smart Keyframe Selection',
description="Enable and use the regular keyframe selection.", description="Use the smart keyframe selection.",
value=True, value=True,
uid=[0], uid=[0]
enabled=False, # only method for now, it must always be enabled
), ),
desc.IntParam( desc.GroupAttribute(
name="minFrameStep", name="regularSelection",
label="Min Frame Step", label="Regular Keyframe Selection",
description="Minimum number of frames between two keyframes.", description="Parameters for the regular keyframe selection.\nKeyframes are selected regularly over the sequence with respect to the set parameters.",
value=12, group=None, # skip group from command line
range=(1, 1000, 1), enabled=lambda node: node.selectionMethod.useSmartSelection.value is False,
uid=[0], groupDesc=[
enabled=lambda node: node.regularSelection.useRegularSelection.value desc.IntParam(
name="minFrameStep",
label="Min Frame Step",
description="Minimum number of frames between two keyframes.",
value=12,
range=(1, 1000, 1),
uid=[0],
enabled=lambda node: node.regularSelection.enabled
),
desc.IntParam(
name="maxFrameStep",
label="Max Frame Step",
description="Maximum number of frames between two keyframes. Ignored if equal to 0.",
value=0,
range=(0, 1000, 1),
uid=[0],
enabled=lambda node: node.regularSelection.enabled
),
desc.IntParam(
name="maxNbOutFrames",
label="Max Nb Output Frames",
description="Maximum number of output frames (0 = no limit).\n"
"'minFrameStep' and 'maxFrameStep' will always be respected, so combining them with this parameter\n"
"might cause the selection to stop before reaching the end of the input sequence(s).",
value=0,
range=(0, 10000, 1),
uid=[0],
enabled=lambda node: node.regularSelection.enabled
),
],
), ),
desc.IntParam( desc.GroupAttribute(
name="maxFrameStep", name="smartSelection",
label="Max Frame Step", label="Smart Keyframe Selection",
description="Maximum number of frames between two keyframes. Ignored if equal to 0.", description="Parameters for the smart keyframe selection.\nKeyframes are selected based on their sharpness and optical flow scores.",
value=0, group=None, # skip group from command line
range=(0, 1000, 1), enabled=lambda node: node.selectionMethod.useSmartSelection.value,
uid=[0], groupDesc=[
enabled=lambda node: node.regularSelection.useRegularSelection.value desc.FloatParam(
), name="pxDisplacement",
desc.IntParam( label="Pixel Displacement",
name="maxNbOutFrames", description="The percentage of pixels in the frame that need to have moved since the last keyframe to be considered for the selection",
label="Max Nb Output Frames", value=3.0,
description="Maximum number of output frames (0 = no limit).\n" range=(0.0, 100.0, 1.0),
"'minFrameStep' and 'maxFrameStep' will always be respected, so combining them with this parameter\n" uid=[0],
"might cause the selection to stop before reaching the end of the input sequence(s).", enabled=lambda node: node.smartSelection.enabled
value=0, ),
range=(0, 10000, 1), desc.IntParam(
uid=[0], name="minNbOutFrames",
enabled=lambda node: node.regularSelection.useRegularSelection.value label="Min Nb Output Frames",
), description="Minimum number of frames selected to be keyframes.",
], value=10,
range=(1, 100, 1),
uid=[0],
enabled=lambda node: node.smartSelection.enabled
),
desc.IntParam(
name="maxNbOutFrames",
label="Max Nb Output Frames",
description="Maximum number of frames selected to be keyframes.",
value=2000,
range=(1, 10000, 1),
uid=[0],
enabled=lambda node: node.smartSelection.enabled
),
desc.IntParam(
name="rescaledWidth",
label="Rescaled Frame's Width",
description="Width, in pixels, of the frame after a rescale. Aspect ratio will be preserved. No rescale will be performed if equal to 0.",
value=720,
range=(0, 4000, 1),
uid=[0],
enabled=lambda node: node.smartSelection.enabled,
advanced=True
),
desc.IntParam(
name="sharpnessWindowSize",
label="Sharpness Window Size",
description="The size, in pixels, of the sliding window used to evaluate a frame's sharpness.",
value=200,
range=(1, 10000, 1),
uid=[0],
enabled=lambda node: node.smartSelection.enabled,
advanced=True
),
desc.IntParam(
name="flowCellSize",
label="Optical Flow Cell Size",
description="The size, in pixels, of the cells within a frame in which the optical flow scores is evaluated.",
value=90,
range=(10, 2000, 1),
uid=[0],
enabled=lambda node: node.smartSelection.enabled,
advanced=True
),
]
)
]
), ),
desc.ChoiceParam( desc.ChoiceParam(
name="verboseLevel", name="verboseLevel",