Compute transformation matrices for all bones Args: pose_params: Dict of bone_name -> tensor [batch_size, 4, 4] (rotation

Create first frame with BOTH red mask overlay AND grid lines Returns: (output_path, grid_rows, grid_cols)

Create first frame with BOTH red mask overlay AND grid lines Returns: (output_path, grid_rows, grid_cols)

Extract pose parameters organized by bone name. Args: armature_data: List of frame dictionaries Returns: Dictionary map

Extract optical flow for an entire video sequence. Args: video_frames: Video tensor [B, T, C, H, W] or [T, C, H, W]

Extract VideoJAM motion representation for an entire video sequence. This performs: 1. RAFT flow extraction between consecut

r"""Forward method of the `CogVideoXUpBlock3D` class.

Convert to float [0,1], simple inverse gamma (approx linearize)

RoPE for video tokens with 3D structure. Args: embed_dim: (`int`): The embedding dimension size, corresponding to hidden_size_he

Get all objects in the scene. Returns: A list of all objects in the scene.

Get grid row, col from canvas position

Get bounding box size of object

Convert HSV color space to RGB using PyTorch operations. Args: hsv: HSV tensor [..., 3, H, W] with values in [0, 1] hsv

Load JSON config file via dialog

Load HUMOTO sequence from pickle file. Args: sequence_path: Path to sequence directory y_up: Convert from Z-up to Y-up coord

Load physics configuration from JSON file.

Create VLM prompt for analyzing video with primary mask

Create VLM prompt for analyzing video with primary mask

Convert 4x4 transformation matrix to quaternion [w, x, y, z, tx, ty, tz].

Process loaded config

Process a single folder containing rgb_full.mp4, rgb_removed.mp4, and mask.mp4. Creates a new quadmask.mp4 file.

Process a single folder containing mask video. Creates a new mask_grid.mkv file with hybrid grid-aligned masks.

Convert quaternion to 4x4 transformation matrix. Quaternion format: [w, x, y, z, tx, ty, tz] or [w, x, y, z]

Read all frames from a video

Calls the scheduler's `set_timesteps` method and retrieves timesteps from the scheduler after the call. Handles custom timesteps. Any kwargs

Save current points and move to next video

Save the scene as a BLEND file. Args: blend_output_path: The path to save the BLEND file. Returns: None

Save physics configuration.

Save the scene as an FBX file. Args: fbx_output_path: The path to save the FBX file. Returns: None

Save the scene as a GLB file. Args: glb_output_path: The path to save the GLB file. Returns: None

Save RGB motion representation as a video file using high-quality encoding. Args: motion_rgb: Motion RGB tensor [T, 3, H, W] or [B,

Save conversion progress.

Segment object using text prompt - returns boolean mask

Segment video using point prompts (single or multi-frame). Args: video_path: Path to input video points: Lis

if alpha == 0 or None, alpha is rank (no scaling).

Extract video frames as JPG files

Lazy import kaolin with caching

Write masks to video file

Old model hash used by sd-webui-additional-networks for .safetensors format files

New model hash used by sd-webui-additional-networks for .safetensors format files

Animate human mesh by creating shape keys for each frame. Args: human_obj: Blender object with the human mesh all_vertices:

Animate object transformations.

Apply matte material properties to character meshes. Makes characters look realistic and non-shiny.

Apply object along trajectory path across frames. Args: obj_mask: Object mask from first_appears_frame trajectory_points: Li

Applies the object-level scale of a selected armature and its child meshes by baking a new, corrected animation clip. This script perfor

Calculate camera position and look-at point from hip position. Camera will be positioned in FRONT of the human, not behind. Args:

Calculate grid size maintaining aspect ratio with minimum cells on shorter side. Args: width: Video width in pixels height:

Calculate grid dimensions for square cells

Calculate grid dimensions for square cells

Calculate grid dimensions for square cells

Calculate grid dimensions to make square cells