nemo_curator.stages.video.filtering.motion_filter

Module Contents

Classes

Name	Description
`MotionFilterStage`	Stage for filtering video clips based on motion score.
`MotionVectorDecodeStage`	Stage for decoding motion vector information from video files.

API

class nemo_curator.stages.video.filtering.motion_filter.MotionFilterStage(
    score_only: bool = False,
    global_mean_threshold: float = 0.00098,
    per_patch_min_256_threshold: float = 1e-06,
    num_gpus_per_worker: float = 0,
    motion_filter_batch_size: int = 256,
    verbose: bool = False,
    name: str = 'motion_filter'
)

Dataclass

Bases: ProcessingStage[VideoTask, VideoTask]

Stage for filtering video clips based on motion score.

This class processes video clips through a series of steps including motion score computation and filtering based on thresholds.

global_mean_threshold

float = 0.00098

motion_filter_batch_size

int = 256

name

str = 'motion_filter'

num_gpus_per_worker

float = 0

per_patch_min_256_threshold

float = 1e-06

score_only

bool = False

verbose

bool = False

nemo_curator.stages.video.filtering.motion_filter.MotionFilterStage.__post_init__() -> None

nemo_curator.stages.video.filtering.motion_filter.MotionFilterStage.inputs() -> tuple[list[str], list[str]]

nemo_curator.stages.video.filtering.motion_filter.MotionFilterStage.outputs() -> tuple[list[str], list[str]]

nemo_curator.stages.video.filtering.motion_filter.MotionFilterStage.process(
    task: nemo_curator.tasks.video.VideoTask
) -> nemo_curator.tasks.video.VideoTask

class nemo_curator.stages.video.filtering.motion_filter.MotionVectorDecodeStage(
    num_cpus_per_worker: float = 6.0,
    verbose: bool = False,
    target_fps: float = 2.0,
    target_duration_ratio: float = 0.5,
    name: str = 'motion_vector_decoding'
)

Dataclass

Bases: ProcessingStage[VideoTask, VideoTask]

Stage for decoding motion vector information from video files.

This class processes video files through a series of steps including decoding, filtering by side length, and storing the results in the task.

name

str = 'motion_vector_decoding'

num_cpus_per_worker

float = 6.0

target_duration_ratio

float = 0.5

target_fps

float = 2.0

verbose

bool = False

nemo_curator.stages.video.filtering.motion_filter.MotionVectorDecodeStage.__post_init__() -> None

nemo_curator.stages.video.filtering.motion_filter.MotionVectorDecodeStage.inputs() -> tuple[list[str], list[str]]

nemo_curator.stages.video.filtering.motion_filter.MotionVectorDecodeStage.outputs() -> tuple[list[str], list[str]]

nemo_curator.stages.video.filtering.motion_filter.MotionVectorDecodeStage.process(
    task: nemo_curator.tasks.video.VideoTask
) -> nemo_curator.tasks.video.VideoTask