stages.video.clipping.transnetv2_extraction

Module Contents

Classes

TransNetV2ClipExtractionStage

Stage for extracting video clips using TransNetV2.

API

class stages.video.clipping.transnetv2_extraction.TransNetV2ClipExtractionStage

Bases: nemo_curator.stages.base.ProcessingStage[nemo_curator.tasks.video.VideoTask, nemo_curator.tasks.video.VideoTask]

Stage for extracting video clips using TransNetV2.

This class processes video clips through a series of steps including shot detection, scene filtering, and clip assignment.

crop_s: float | None

0.5

entire_scene_as_clip: bool

True

gpu_memory_gb: int

10

inputs() → tuple[list[str], list[str]]

Define stage input requirements.

Returns (tuple[list[str], list[str]]): Tuple of (required_attributes, required_columns) where: - required_top_level_attributes: List of task attributes that must be present - required_data_attributes: List of attributes within the data that must be present

limit_clips: int

None

max_length_mode: Literal[truncate, stride]

‘stride’

max_length_s: float | None

10.0

min_length_s: float | None

2.0

model_dir: str

None

outputs() → tuple[list[str], list[str]]

Define stage output specification.

Returns (tuple[list[str], list[str]]): Tuple of (output_attributes, output_columns) where: - output_top_level_attributes: List of task attributes this stage adds/modifies - output_data_attributes: List of attributes within the data that this stage adds/modifies

process(

task: nemo_curator.tasks.video.VideoTask,

) → nemo_curator.tasks.video.VideoTask

Process a task and return the result. Args: task (X): Input task to process Returns (Y | list[Y]): - Single task: For 1-to-1 transformations - List of tasks: For 1-to-many transformations (e.g., readers) - None: If the task should be filtered out

setup(

worker_metadata: nemo_curator.backends.base.WorkerMetadata | None = None,

) → None

Setup method called once before processing begins. Override this method to perform any initialization that should happen once per worker. Args: worker_metadata (WorkerMetadata, optional): Information about the worker (provided by some backends)

setup_on_node(

node_info: nemo_curator.backends.base.NodeInfo,

worker_metadata: nemo_curator.backends.base.WorkerMetadata,

) → None

Download TransNetV2 weights on the node.

threshold: float

0.4

verbose: bool

False