nemo_curator.stages.video.preview.preview

Module Contents

Classes

Name	Description
`PreviewStage`	Stage that generates webp previews from video clips.

API

class nemo_curator.stages.video.preview.preview.PreviewStage(
    target_fps: float = 1.0,
    target_height: int = 240,
    verbose: bool = False,
    num_cpus_per_worker: float = 4.0,
    compression_level: int = 6,
    quality: int = 50
)

Dataclass

Bases: ProcessingStage[VideoTask, VideoTask]

Stage that generates webp previews from video clips.

This class processes video clips through a series of steps including reading, generating webp previews, and writing to storage.

compression_level

int = 6

num_cpus_per_worker

float = 4.0

quality

int = 50

target_fps

float = 1.0

target_height

int = 240

verbose

bool = False

nemo_curator.stages.video.preview.preview.PreviewStage.__post_init__() -> None

nemo_curator.stages.video.preview.preview.PreviewStage._generate_preview(
    window: nemo_curator.tasks.video._Window
) -> None

Generate webp preview for a video window.

Parameters:

window

_Window

Window containing video data to generate preview for.

nemo_curator.stages.video.preview.preview.PreviewStage.inputs() -> tuple[list[str], list[str]]

nemo_curator.stages.video.preview.preview.PreviewStage.outputs() -> tuple[list[str], list[str]]

nemo_curator.stages.video.preview.preview.PreviewStage.process(
    task: nemo_curator.tasks.video.VideoTask
) -> nemo_curator.tasks.video.VideoTask