Remove all objects from the scene.

Transfer color distribution from of sc, referred to dc. Args: sc (numpy.ndarray): input image to be transfered. dc (numpy.nd

Combine black and grey mask frames. Rules: - black=0, grey=255 → 0 (primary object only) - black=255, grey=127 → 127 (affected objec

Compute the positions of all joints in world space Args: pose_params: Dict of bone_name -> tensor [batch_size, 4, 4] (rot

Convert quad mask to hybrid grid-aligned mask. Process: 1. Grid-align 127 (grey): cell gets 127 if ANY pixel is 127, else 255 2. Kee

Convert one sequence for a specific character. Runs the 4-step conversion process.

Create visualization of first frame with red overlay on primary object Args: first_frame_path: Path to first_frame.jpg black_mask

Create visualization of first frame with red overlay on primary object Args: first_frame_path: Path to first_frame.jpg black_mask

Create a ground plane with optional PBR textures or chessboard pattern. Args: size: Size of the plane y_up: Whether using Y-up co

Create human mesh from pre-computed vertices and faces. Args: vertices: numpy array [num_verts, 3] - first frame vertices fa

Create RGB visualization of mask with color coding and RGB background

Create gridded frame samples at multiple time points in video. Helps VLM see objects that appear mid-video with grid reference. Args:

Create gridded frame samples at multiple time points in video. Helps VLM see objects that appear mid-video with grid reference. Args:

Create quad-mask from tri-mask and RGB frames. Args: mask_frame: Original mask frame (numpy array) rgb_full_frame: Full RGB

Build quad-mask using 4 renders: 0 (black) : human only (no scene change underneath) 63 (dark gray) : overlap (human coverin

Create video from frames using ffmpeg.

Create grayscale video from frames using ffmpeg.

Create 4 walls around the scene with optional PBR textures. Args: size: Size of the room (walls will be at +/- size/2) height: He

returns either a context list that includes one that will disable zero.Init or an empty context list

returns either a context list that includes one that will disable zero.Init or an empty context list

Dilate mask for proximity checking

Dilate mask to create proximity region

Draw grid lines on mask canvas

r""" Encodes the prompt into text encoder hidden states. Args: prompt (`str` or `List[str]`, *optional*):

r""" Encodes the prompt into text encoder hidden states. Args: prompt (`str` or `List[str]`, *optional*):

r""" Encodes the prompt into text encoder hidden states. Args: prompt (`str` or `List[str]`, *optional*):

Extract optical flow between two consecutive frames. Args: frame1: First frame [B, C, H, W] in range [0, 1] or [0, 255]

Extract all frames from a video using ffmpeg.

Extract hip positions from armature data for camera placement. Args: armature_data: List of frame bone dictionaries bone_nam

Extract mesh vertices and faces from objects. Works in both object mode and edit mode.

Filter mask to only include regions near primary mask

Filter masks to only include regions near primary mask

Find all objects that need trajectory input

Find the pass 1 video for a given video name. Args: video_name: Base name (e.g., "ball-play") pass1_dir: Directory containin

Find all sequences with pre-computed vertices.

Find all sequences with pre-computed vertices.

Find all folders containing the required video files.

Find all folders containing mask videos.

All objects done

Stable flow->RGB with fixed pixel clip for consistent visualization. NOTE: This is for VISUALIZATION, not VideoJAM training! For VideoJA

Convert optical flow to RGB motion representation following VideoJAM paper specification. This is NOT a visualization - it's a weak, normali

Generate warped noise from a video using make_warped_noise.py. Args: video_path: Path to input video output_dir: Directory t

Get all sequence directories.

given an array of points *a*, create a curve through those points. *rad* is a number between 0 and 1 to steer the distance of cont

given an array of points *a*, create a curve through those points. *rad* is a number between 0 and 1 to steer the distance of cont

Get realistic color for an object. Args: obj_name: Name of the object character_name: Name of character ('sophie' or 'remy')

Get frame at current_frame_idx from current video

Get a list of bright, vibrant colors for objects. These colors are distinct from dark human colors (dark grey, dark blue, dark purple). Expand

Get physics type for a specific object in a sequence. Returns: 'active': object should move/fall 'passive': object should st

Randomly select a camera motion type. Args: seed: Random seed for reproducible selection Returns: str: One of 'static',

create n random points in the unit square, which are *mindst* apart, then scale them.

create n random points in the unit square, which are *mindst* apart, then scale them.

Get appropriate texture path for an object, or None if it should use color. Args: obj_name: Name of the object Returns:

Get video width and height using ffprobe.

Convert grid cells to mask

Convert pixel mask to gridified mask

Gridify masks: if ANY pixel in grid cell → ENTIRE cell = True Args: masks: List of boolean masks (one per frame) grid_rows,

Load warped noise and resize to match latent dimensions. Args: noise_path: Path to warped noise .npy file target_shape: (lat