nemo_curator.stages.video.clipping.clip_frame_extraction

Module Contents

Classes

Name	Description
`ClipFrameExtractionStage`	Stage for extracting frames from video clips.

API

class nemo_curator.stages.video.clipping.clip_frame_extraction.ClipFrameExtractionStage(
    extraction_policies: tuple[nemo_curator.utils.decoder_utils.FrameExtractionPolicy, ...] = (FrameExtractionPolicy.sequ...,
    extract_purposes: list[nemo_curator.utils.decoder_utils.FramePurpose] | None = None,
    target_res: tuple[int, int] | None = None,
    verbose: bool = False,
    num_cpus: int = 3,
    target_fps: list[float | int] | None = None,
    name: str = 'clip_frame_extraction'
)

Dataclass

Bases: ProcessingStage[VideoTask, VideoTask]

Stage for extracting frames from video clips.

This class processes video clips through a series of steps including frame extraction, target frame rate selection, and frame extraction signature creation.

extract_purposes

list[FramePurpose] | None = None

extraction_policies

tuple[FrameExtractionPolicy, ...] = (FrameExtractionPolicy.sequence,)

name

str = 'clip_frame_extraction'

num_cpus

int = 3

target_fps

list[float | int] | None = None

target_res

tuple[int, int] | None = None

verbose

bool = False

nemo_curator.stages.video.clipping.clip_frame_extraction.ClipFrameExtractionStage.__post_init__() -> None

nemo_curator.stages.video.clipping.clip_frame_extraction.ClipFrameExtractionStage.inputs() -> tuple[list[str], list[str]]

nemo_curator.stages.video.clipping.clip_frame_extraction.ClipFrameExtractionStage.lcm_multiple(
    fps: list[float | int]
) -> float | int

Compute LCM of a list of fps targets.

nemo_curator.stages.video.clipping.clip_frame_extraction.ClipFrameExtractionStage.outputs() -> tuple[list[str], list[str]]

nemo_curator.stages.video.clipping.clip_frame_extraction.ClipFrameExtractionStage.process(
    task: nemo_curator.tasks.video.VideoTask
) -> nemo_curator.tasks.video.VideoTask

nemo_curator.stages.video.clipping.clip_frame_extraction.ClipFrameExtractionStage.setup(
    worker_metadata: nemo_curator.backends.base.WorkerMetadata | None = None
) -> None