Skip to content

Examples

This page provides practical examples for common tasks with sleap-io. Each example includes working code that you can copy and adapt for your needs.

Prerequisites

All examples assume you have sleap-io installed. See the Installation Guide for options including uv, pip, and development setup.

Quick start with uv (recommended):

# Run any example script directly (no install needed)
uv run --with sleap-io example.py

This automatically handles dependencies without needing to manage environments.

Most examples use import sleap_io as sio for brevity.

Basics

Create labels from raw data

Build a complete Labels dataset programmatically from a Skeleton, Video, and Instance objects.

create_labels.py
import sleap_io as sio
import numpy as np

# Create skeleton
skeleton = sio.Skeleton(
    nodes=["head", "thorax", "abdomen"],
    edges=[("head", "thorax"), ("thorax", "abdomen")]
)

# Create video
video = sio.load_video("test.mp4")

# Create instance from numpy array
instance = sio.Instance.from_numpy(
    points_data=np.array([
        [10.2, 20.4],  # head
        [5.8, 15.1],   # thorax
        [0.3, 10.6],   # abdomen
    ]),
    skeleton=skeleton
)

# Create labeled frame
lf = sio.LabeledFrame(video=video, frame_idx=0, instances=[instance])

# Create labels
labels = sio.Labels(videos=[video], skeletons=[skeleton], labeled_frames=[lf])

# Save
labels.save("labels.slp")
Creating predicted instances

To create predictions with confidence scores, include scores as the third column of the points array:

predicted = sio.PredictedInstance.from_numpy(
    np.array([
        [10.2, 20.4, 0.9],   # head (x, y, score)
        [5.8, 15.1, 0.8],    # thorax
        [0.3, 10.6, 0.7],    # abdomen
    ]),
    skeleton=skeleton,
    score=0.85,  # instance-level confidence
)

See also

Convert labels to raw arrays

Extract pose data as NumPy arrays for analysis or visualization using Labels.numpy().

labels_to_numpy.py
import sleap_io as sio

labels = sio.load_slp("tests/data/slp/centered_pair_predictions.slp")

# Convert predictions to point coordinates in a single array
trx = labels.numpy()
n_frames, n_tracks, n_nodes, xy = trx.shape
assert xy == 2  # x and y coordinates

# Convert to array with confidence scores appended
trx_with_scores = labels.numpy(return_confidence=True)
n_frames, n_tracks, n_nodes, xy_score = trx_with_scores.shape 
assert xy_score == 3  # x, y, and confidence score
Expected output shapes

For a dataset with 100 frames, 2 tracks, and 3 nodes:

  • Without scores: (100, 2, 3, 2)
  • With scores: (100, 2, 3, 3)

See also

Labels.numpy: Full documentation of array conversion options

Annotation types

Each new v0.7.0 annotation type (bounding boxes, centroids, identities, and 3-D instances) has a User* variant for ground-truth annotations and a Predicted* variant carrying confidence scores. All of them live inside LabeledFrame and are appended with the same type-dispatching LabeledFrame.append.

Bounding box detections

Create, inspect, and convert BoundingBox annotations for detection and tracking workflows.

bbox_basics.py
import sleap_io as sio

# Construct from corners or from top-left + size
bbox = sio.UserBoundingBox(x1=10, y1=20, x2=100, y2=200, category="mouse")
bbox = sio.UserBoundingBox.from_xywh(10, 20, 90, 180)  # same box

# Query geometric properties
bbox.xyxy         # (10, 20, 100, 200)  (ints preserved from int inputs)
bbox.xywh         # (10, 20, 90, 180)
bbox.area         # 16200
bbox.centroid_xy  # (55.0, 110.0)

# Predictions carry a score (and optional tracking_score)
pred_bbox = sio.PredictedBoundingBox(
    x1=10, y1=20, x2=100, y2=200, score=0.95, tracking_score=0.88,
)

# Convert to related annotation types
roi = bbox.to_roi()
mask = bbox.to_mask(height=480, width=640)

Attach a box to a frame and save it into SLP (format v2.0+) — bboxes also round-trip through COCO detection, Ultralytics detection, GeoJSON, and JABS when the dataset carries the metadata those formats require.

bbox_save.py
import sleap_io as sio

video = sio.Video.from_filename("clip.mp4", open=False)
lf = sio.LabeledFrame(video=video, frame_idx=0)
lf.append(sio.UserBoundingBox(x1=10, y1=20, x2=100, y2=200, category="mouse"))

labels = sio.Labels(labeled_frames=[lf], videos=[video])
labels.save("boxes.slp")

See also

Centroid tracking and TrackMate import

Centroid annotations are lightweight (x, y) points used by TrackMate, SORT, ByteTrack, and similar trackers. They interconvert with full pose Instance objects.

centroid_basics.py
import sleap_io as sio

track = sio.Track(name="mouse_A")

# User and predicted variants
c = sio.UserCentroid(x=100.5, y=200.3, track=track)
p = sio.PredictedCentroid(x=50.0, y=60.0, score=0.95, source="trackmate")

# Convert a pose Instance into a single-point centroid
centroid = sio.Centroid.from_instance(pose_instance, method="center_of_mass")

# And back: produces a single-node Instance on CENTROID_SKELETON
inst = centroid.to_instance()

# Convenience accessors on full pose Instances
xy = pose_instance.centroid_xy        # (x, y) tuple
c2 = pose_instance.to_centroid()      # shortcut for Centroid.from_instance(pose_instance)

Import TrackMate *_spots.csv exports straight into a Labels object — either with the explicit loader or by letting sio.load_file auto-detect the format from the CSV header:

load_trackmate.py
import sleap_io as sio

# Explicit loader (use when you also want to pass edges or other options)
labels = sio.load_trackmate("spots.csv", video="experiment.tif")

# Auto-detect from CSV headers (falls back to DLC and generic CSV readers)
labels = sio.load_file("spots.csv")

print(len(labels.centroids), "centroid detections")

See also

Cross-session identity and 3-D instances

Identity identifies the same animal across multiple recording sessions, distinct from per-video Track. Instance3D / PredictedInstance3D carry structured triangulated keypoints, and InstanceGroup ties multi-view 2-D instances together with their 3-D reconstruction.

identity_and_3d.py
import numpy as np
import sleap_io as sio

skel = sio.Skeleton(nodes=["head", "tail"], edges=[("head", "tail")])

# Persistent per-animal identity (matched by name; a color, if wanted, is just
# a conventional free-form metadata key, e.g. metadata={"color": "#e6194b"})
mouse_a = sio.Identity(name="mouse_A")

# 3-D keypoint storage, aligned to a skeleton
pts = np.array([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]])
inst_3d = sio.Instance3D(points=pts, skeleton=skel)
pred_3d = sio.PredictedInstance3D(
    points=pts, skeleton=skel, point_scores=np.array([0.9, 0.8]),
)

# Group per-camera 2-D detections with the reconstructed 3-D instance
cam1 = sio.Camera(name="cam1")
cam2 = sio.Camera(name="cam2")
inst_2d_cam1 = sio.Instance.from_numpy(np.zeros((2, 2)), skeleton=skel)
inst_2d_cam2 = sio.Instance.from_numpy(np.zeros((2, 2)), skeleton=skel)
group = sio.InstanceGroup(
    instance_by_camera={cam1: inst_2d_cam1, cam2: inst_2d_cam2},
    instance_3d=inst_3d,
    identity=mouse_a,
)

# Identities and recording sessions hang off Labels directly
labels = sio.Labels(identities=[mouse_a])

Identities persist into SLP format v2.5+ and Instance3D into v1.9+, so round-tripping between sleap-io and sleap-io.js / luc3d is lossless.

See also

3D model: Camera, CameraGroup, RecordingSession, FrameGroup, InstanceGroup, Identity, Instance3D.

Behavior segmentation with events (HITL)

An Event is the first frame-spanning annotation: a (video, start_frame, end_frame, type) interval for behavior bouts, stimulus epochs, review flags — anything with a temporal extent. Events live on Labels.events (not on a LabeledFrame), reference a Track / Identity as subject / target, and draw their vocabulary from an EventType catalog. This is the data model for human-in-the-loop behavior segmentation: a model proposes PredictedEvents and a human accepts/edits them as UserEvents.

events_hitl.py
import sleap_io as sio

labels = sio.load_slp("session.slp")
video = labels.videos[0]
mouse1, mouse2 = labels.tracks[:2]

# Define the ethogram once (a color is just conventional free-form metadata).
attack = sio.EventType(name="attack", metadata={"color": "#e6194b"})

# A model proposes bouts with framewise + scalar confidence (both optional).
labels.events.append(
    sio.PredictedEvent(
        type=attack, video=video, start_frame=100, end_frame=140,   # inclusive
        subject=mouse1, target=mouse2,                              # directed
        scores=[0.6] * 41, score=0.74,
    )
)
# A human accepts/edits it as ground truth (a bare string auto-promotes to an
# EventType(name=...); target=None means non-directed / "self").
labels.events.append(
    sio.UserEvent(type="rear", video=video, start_frame=120, subject=mouse1)
)

# Query what is happening, when, and to whom.
labels.get_events(type="attack", predicted=True)     # model proposals to review
labels.events_at(video, 130)                         # events covering frame 130
[e for e in labels.get_events(subject=mouse1) if e.is_directed]

# Persists to SLP format 2.6+ (additive /event_types + /events groups).
labels.save("session_scored.slp")
scored = sio.load_slp("session_scored.slp")
scored.events, scored.event_types

See also

  • Events model — full API, frame convention, and SLP format details
  • Poses / Embeddings — the Track / Identity catalogs events reference

Reading and writing GeoJSON ROIs

ROIs are serializable to the movement-compatible GeoJSON format used by Shapely, GeoPandas, QGIS, and QuPath. Each ROI also implements the Python __geo_interface__ protocol so it plugs straight into any geopandas.GeoDataFrame.

geojson_rois.py
import sleap_io as sio

labels = sio.load_slp("labels.slp")

# Dump every ROI (static and frame-level) to GeoJSON
sio.save_geojson(labels.rois, "regions.geojson")

# Load an externally drawn region set
rois = sio.load_geojson("regions.geojson")
first = rois[0]
first.__geo_interface__   # standard GeoJSON Feature dict

# Universal loader auto-detects the .geojson extension
labels = sio.load_file("regions.geojson")

See also

Adding annotations to frames (type dispatch)

LabeledFrame.append() routes each annotation to the correct per-type list based on its runtime type. The same method handles every annotation kind — you never have to touch lf.bboxes, lf.centroids, etc. by hand for appends.

frame_append_dispatch.py
import numpy as np
import sleap_io as sio
from shapely.geometry import box

skel = sio.Skeleton(nodes=["head"], edges=[])
video = sio.Video.from_filename("clip.mp4", open=False)
lf = sio.LabeledFrame(video=video, frame_idx=0)

# Each append() routes by type:
lf.append(sio.Instance.from_numpy(np.zeros((1, 2)), skeleton=skel))  # → lf.instances
lf.append(sio.UserBoundingBox(x1=0, y1=0, x2=10, y2=10))             # → lf.bboxes
lf.append(sio.UserCentroid(x=5, y=5))                                # → lf.centroids
lf.append(sio.UserSegmentationMask.from_numpy(np.zeros((8, 8), bool)))  # → lf.masks
lf.append(sio.UserLabelImage.from_numpy(np.zeros((8, 8), int)))      # → lf.label_images
lf.append(sio.UserROI(geometry=box(0, 0, 10, 10)))                   # → lf.rois

# Per-frame accessors are plain lists you can iterate or index:
print(len(lf.instances), len(lf.bboxes), len(lf.centroids))
print(len(lf.masks), len(lf.label_images), len(lf.rois))

# Flat read-only views across the whole Labels object
labels = sio.Labels(labeled_frames=[lf], videos=[video])
labels.bboxes      # every bbox across every frame
labels.centroids   # every centroid across every frame
labels.static_rois # video-level ROIs (separate from frame-level lf.rois)

See also

Regions → Working with annotations in frames for more context on the flat-vs-nested views.

Fast O(1) frame and track lookups

Labels.get_frame and Labels.get_track_annotations replace O(n) linear scans with O(1) dictionary lookups backed by lazy indices. Use them when iterating over large datasets or cross-referencing by frame / track.

fast_lookups.py
import sleap_io as sio

labels = sio.load_slp("predictions.slp")
video = labels.videos[0]
track = labels.tracks[0]

# O(1) frame lookup (returns None if the frame is missing)
lf = labels.get_frame(video, frame_idx=42)
if lf is not None:
    print(len(lf.instances), "instances on frame 42")

# O(1) per-track annotation view (sorted by frame_idx)
annotations = labels.get_track_annotations(video, track)
for ann in annotations[:5]:
    print(ann.frame_idx, type(ann).__name__)

# Existing convenience methods now use the same indices internally
labels.find(video, frame_idx=42)                    # O(1)
labels.get_centroids(video=video, frame_idx=0)      # O(1)
labels.get_bboxes(video=video, frame_idx=0)         # O(1)

# If you mutate labels.labeled_frames in place, force a rebuild
labels.labeled_frames.append(sio.LabeledFrame(video=video, frame_idx=99))
labels.reindex()

When to call reindex()

The indices rebuild automatically when you use the public Labels APIs. You only need reindex() after mutating labels.labeled_frames or the annotation lists in place (which bypasses the invalidation hooks).

Resolving a video by path or foreign instance

Video objects compare by identity, so a Video you create yourself (e.g. with sio.load_video) is not recognized by Labels lookups even if it points at the same file as one already in the project. Labels.match_video resolves a foreign Video — or a plain filename — to the canonical instance, and find, extract, __getitem__, numpy, and the get_* family all canonicalize their video argument through it automatically.

resolve_video.py
import sleap_io as sio

labels = sio.load_slp("predictions.slp")

# A freshly loaded Video is a different object than the one in the project.
foreign = sio.load_video("predictions_video.mp4")

# These all work now (previously returned [] / raised IndexError):
labels.find(foreign)                      # by foreign Video instance
labels.find("predictions_video.mp4")      # by filename
labels["predictions_video.mp4"]           # __getitem__ by filename
labels.extract([("predictions_video.mp4", 0)])

# Resolve explicitly to the canonical Video stored on the project:
canonical = labels.match_video(foreign)   # -> Video, or None if no match

Matching strategy

match_video uses a tiered cascade by default: a definitive match (same underlying file, or identical path) wins, and only if none is found does it fall back to basename matching. Ambiguous matches raise ValueError. Pass method= ("path", "basename", "content", ...) for explicit control.

Format conversion

Load and save in different formats

Convert between supported formats with automatic format detection using load_file() and Labels.save().

format_conversion.py
import sleap_io as sio

# Load from SLEAP file
labels = sio.load_file("predictions.slp")

# Save to NWB file
labels.save("predictions.nwb")

Tip

sleap-io automatically detects the format from the file extension. Supported formats include .slp, .nwb, .json (COCO/Label Studio), .labelstudio.json, .h5 (JABS), and .mat (LEAP). Use format="coco" to explicitly save as COCO format.

See also

  • Labels.save: Save method with format options
  • Formats: Complete list of supported formats

Working with NWB files

Neurodata Without Borders (NWB) provides a standardized format for neurophysiology data. sleap-io offers comprehensive NWB support with automatic format detection.

nwb_basic.py
import sleap_io as sio

# Load any NWB file - automatically detects if it contains
# annotations (PoseTraining) or predictions (PoseEstimation)
labels = sio.load_nwb("pose_data.nwb")

# Save with automatic format detection
# Uses "annotations" if data has user labels, "predictions" otherwise
sio.save_nwb(labels, "output.nwb")

# Force specific format
sio.save_nwb(labels, "training.nwb", nwb_format="annotations")
sio.save_nwb(labels, "inference.nwb", nwb_format="predictions")

# Export with embedded video frames for sharing complete datasets
sio.save_nwb(labels, "dataset_export.nwb", nwb_format="annotations_export")

Format auto-detection

The harmonization layer automatically determines the appropriate format:

  • Annotations: Used when data contains user-labeled instances (training data)
  • Predictions: Used when data contains only predicted instances (inference results)
  • Annotations Export: Use explicitly to create self-contained files with embedded video frames

Save training data with rich metadata

Include detailed experimental metadata when saving training annotations.

nwb_metadata.py
import sleap_io as sio

labels = sio.load_file("labels.slp")

# Basic save with default metadata
sio.save_nwb(labels, "training_data.nwb", nwb_format="annotations")
Advanced: custom NWB metadata (internal API)

The public save_nwb() uses default metadata. To customize session descriptions, subject info, and other NWB fields, use the internal save_labels() function. This is not part of the public API and may change between versions.

from sleap_io.io.nwb_annotations import save_labels

save_labels(
    labels,
    "training_data.nwb",
    session_description="Mouse skilled reaching task - training dataset",
    identifier="mouse_01_session_03_annotations",
    session_start_time="2024-01-15T09:30:00",
    annotator="John Doe",
    nwb_kwargs={
        "session_id": "session_003",
        "experimenter": ["John Doe", "Jane Smith"],
        "lab": "Motor Control Lab",
        "institution": "University of Example",
        "experiment_description": "Skilled reaching task with food pellet reward",
        "subject": {
            "subject_id": "mouse_01",
            "age": "P90",
            "sex": "M",
            "species": "Mus musculus",
            "strain": "C57BL/6J",
            "weight": "25g",
        },
    },
)

Metadata best practices

Include as much metadata as possible for reproducibility:

  • Experimental protocol details
  • Subject information
  • Recording conditions
  • Annotator identity for tracking labeling provenance

Export dataset with embedded videos

Create self-contained NWB files with video frames for sharing complete datasets. The simplest approach uses the public API:

nwb_export.py
import sleap_io as sio

labels = sio.load_file("labels.slp")

# Export annotations with embedded video frames to NWB
sio.save_nwb(labels, "dataset_export.nwb", nwb_format="annotations_export")
Advanced: fine-grained export control

For more control over the export process (e.g., custom filenames, multi-subject support), use the internal export functions directly. These are not part of the public API and may change between versions.

from sleap_io.io.nwb_annotations import export_labels, export_labeled_frames

# Export annotations + MJPEG video + frame map to a directory
export_labels(
    labels,
    output_dir="export/",
    mjpeg_filename="annotated_frames.avi",
    frame_map_filename="frame_map.json",
    nwb_filename="pose_training.nwb",
    clean=True,  # Remove empty frames and predictions before export
)

# Or export just labeled frames with provenance tracking
frame_map = export_labeled_frames(
    labels,
    frame_map_path="export/frame_map.json",
    mjpeg_path="export/labeled_frames.avi",
    nwb_path="export/labeled_frames.nwb",
)
print(f"Exported {frame_map.total_frames} frames")

Export contents

  • NWB file: Annotations in PoseTraining format with skeleton definition
  • MJPEG video: Labeled frames re-encoded as a seekable MJPEG video
  • Frame map JSON: Provenance metadata tracking which frames came from which source videos

Convert between NWB and other formats

Use NWB as an interchange format between different pose tracking tools.

nwb_conversion.py
import sleap_io as sio

# Load from DeepLabCut
dlc_data = sio.load_file("dlc_predictions.h5")

# Save as NWB predictions
sio.save_nwb(dlc_data, "dlc_in_nwb.nwb", nwb_format="predictions")

# Load SLEAP training data
sleap_labels = sio.load_file("training.slp")

# Export as NWB with videos for sharing
sio.save_nwb(sleap_labels, "training_export.nwb", nwb_format="annotations_export")

# Convert NWB back to SLEAP format
nwb_labels = sio.load_nwb("training_export.nwb")
nwb_labels.save("converted.slp")

Format preservation

NWB format preserves:

  • Complete skeleton structure with node names
  • Track identities
  • Confidence scores
  • User vs predicted instance types
  • Video metadata (when using annotations_export)

See also

Convert to Ultralytics YOLO format

Export your dataset for use with Ultralytics YOLO models.

ultralytics_export.py
import sleap_io as sio

# Load source labels
labels = sio.load_file("labels.v001.slp")

# Create train/val splits and export as YOLO format
labels_set = labels.make_training_splits(n_train=0.8, n_val=0.2, seed=42)
labels_set.save("yolo_dataset/", format="ultralytics")

# Or export from existing split files
file_dict = {
    "train": "path/to/train.slp",
    "val": "path/to/val.slp",
}
labels_set = sio.load_labels_set(file_dict)
labels_set.save("yolo_dataset/", format="ultralytics")

YOLO export structure

The exported dataset will have the standard YOLO directory structure with train/val splits, images, and label files.

See also

Export to COCO format

Export your dataset for use with mmpose, CVAT, and other COCO-compatible tools.

coco_export.py
import sleap_io as sio

# Load source labels
labels = sio.load_file("labels.slp")

# Export to COCO format
sio.save_coco(labels, "annotations.json")

# Or use save_file with auto-detection from .json extension
labels.save("annotations.json", format="coco")

# Customize export with options
sio.save_coco(
    labels,
    "annotations_binary.json",
    visibility_encoding="binary",  # Use binary (0/1) instead of ternary (0/1/2)
    image_filenames=["frame_001.jpg", "frame_002.jpg", ...]  # Custom filenames
)

mmpose compatibility

The COCO export is fully compatible with mmpose and includes:

  • Required bbox field computed from visible keypoints
  • area field for bounding box area
  • iscrowd field for standard compliance
  • Track IDs via attributes.object_id (CVAT-compatible)
  • 1-based skeleton edge indexing
  • Support for both binary (0/1) and ternary (0/½) visibility encodings

Use cases

COCO export is ideal for:

  • Training pose estimation models with mmpose
  • Annotating data in CVAT and importing to SLEAP
  • Sharing datasets with the broader computer vision community
  • Integration with COCO-compatible evaluation tools

See also

Loading from URLs

load_slp and load_file accept a URL anywhere a local path is accepted, with lazy range-based streaming by default:

import sleap_io as sio

# http/https works with a base install
labels = sio.load_slp("https://example.com/labels.slp")

# Cloud schemes need the [cloud] extra (s3fs / gcsfs / adlfs)
labels = sio.load_slp("s3://my-bucket/labels.slp")

This also covers .pkg.slp embedded-frame streaming, remote media video over http/https, and Google Drive share links. For streaming modes, caching, authentication, security notes, and troubleshooting, see the Remote loading guide.

To instead fetch a remote file to local disk (a curl/wget replacement), use download:

# Download to the current directory, or to a chosen path.
path = sio.download("https://example.com/labels.slp")  # -> ./labels.slp
sio.download("s3://my-bucket/run/video.mp4", "data/")  # -> data/video.mp4

Editing labels data

Fix video paths

Update file paths when moving projects between systems.

fix_paths.py
import sleap_io as sio

# Load labels without trying to open the video files
labels = sio.load_file("labels.v001.slp", open_videos=False)

# Fix paths using prefix replacement
labels.replace_filenames(prefix_map={
    "D:/data/sleap_projects": "/home/user/sleap_projects",
    "C:/Users/sleaper/Desktop/test": "/home/user/sleap_projects",
})

# Save labels with updated paths
labels.save("labels.v002.slp")

Path separators

The prefix map handles path separators automatically, but be consistent with forward slashes (/) for cross-platform compatibility.

Tip

Use open_videos=False when loading to avoid errors from missing videos at the old paths.

See also

Labels.replace_filenames: Additional path manipulation options

Copy labels

Create deep copies of labels with control over video backend behavior.

copy_labels.py
import sleap_io as sio

labels = sio.load_file("labels.slp")

# Default: preserves each video's current open_backend setting
labels_copy = labels.copy()

# Prevent file handles from being created (useful for batch processing)
labels_copy = labels.copy(open_videos=False)

# Force all videos to auto-open when frames are accessed
labels_copy = labels.copy(open_videos=True)

# Filtering done separately (cleaner separation of concerns)
labels_copy = labels.copy()
labels_copy.remove_predictions()
labels_copy.suggestions = []

Non-mutating save

By default, save operations don't mutate the original Labels object:

# Original labels are NOT modified (safer)
labels.save("output.pkg.slp", embed="user")
assert labels.videos[0].filename != "output.pkg.slp"  # Still points to original

# With embed_inplace=True: original labels ARE modified (faster)
labels.save("output.pkg.slp", embed="user", embed_inplace=True)

See also

Labels.copy: Full documentation of copy options

Replace skeleton

Change the skeleton structure while preserving existing annotations.

replace_skeleton.py
import sleap_io as sio

# Load existing labels with skeleton nodes: "head", "trunk", "tti"
labels = sio.load_file("labels.slp")

# Create a new skeleton with different nodes
new_skeleton = sio.Skeleton(["HEAD", "CENTROID", "TAIL_BASE", "TAIL_TIP"])

# Replace skeleton with node correspondence mapping
labels.replace_skeleton(
    new_skeleton,
    node_map={
        "head": "HEAD",
        "trunk": "CENTROID",
        "tti": "TAIL_BASE"
        # "TAIL_TIP" will have NaN values since there's no correspondence
    }
)

# Save with the new skeleton format
labels.save("labels_with_new_skeleton.slp")

Warning

Nodes without correspondence in the node_map will have NaN values in the resulting instances.

Tip

This is particularly useful when converting between different annotation tools or skeleton conventions.

See also

Labels.replace_skeleton: Additional skeleton manipulation options

Update from numpy

Work with pose data as NumPy arrays for filtering or analysis.

numpy_filtering.py
import sleap_io as sio
import numpy as np

labels = sio.load_file("predictions.slp")

# Convert to array — shape: (n_frames, n_tracks, n_nodes, 2)
# In v0.7.0, n_frames == len(video); frames past the last labeled frame are NaN.
trx = labels.numpy()

# Apply temporal smoothing along the frame axis (axis=0)
# This moving average operates independently per track/node/coordinate
kernel = np.ones(5) / 5
trx_smoothed = np.apply_along_axis(
    lambda x: np.convolve(x, kernel, mode="same"), axis=0, arr=trx
)

# Update the labels with smoothed coordinates
labels.update_from_numpy(trx_smoothed)

# Save the smoothed version
labels.save("predictions.smoothed.slp")

Advanced filtering with movement

For more sophisticated temporal filtering (Kalman, median, Savitzky-Golay), check out the movement library which provides purpose-built tools for pose trajectory processing.

Warning

When updating from numpy, the array shape must match the original data structure exactly.

See also

Exporting labels

Save labels with embedded images

Create self-contained label files with embedded video frames.

embed_images.py
import sleap_io as sio

# Load source labels
labels = sio.load_file("labels.v001.slp")

# Save with embedded images for frames with user labeled data and suggested frames
labels.save("labels.v001.pkg.slp", embed="user+suggestions")

Embedding options

  • "user": Only frames with manual annotations
  • "user+suggestions": Manual annotations plus suggested frames
  • "all": All frames with any labels (including predictions)
  • "source": Embed source video if labels were loaded from embedded data

See also

Labels.save: Complete save options including embedding

Advanced embedding options

Progress callback for GUI integration:

embed_with_progress.py
import sleap_io as sio

labels = sio.load_file("labels.slp")

def on_progress(current, total):
    print(f"Embedding frame {current}/{total}")
    return True  # Return False to cancel

labels.save("labels.pkg.slp", embed="user", progress_callback=on_progress)

Cancellation support (e.g., for GUI integration where a user can click "Cancel"):

embed_with_cancel.py
import sleap_io as sio
from sleap_io.io.slp import ExportCancelled

labels = sio.load_file("labels.slp")

cancelled = False  # Set to True from another thread/signal to cancel

def on_progress(current, total):
    return not cancelled  # Return False to cancel the export

try:
    labels.save("output.pkg.slp", embed="user", progress_callback=on_progress)
except ExportCancelled:
    print("Export was cancelled")

Control video embedding for videos without labels:

embed_control.py
# Default: all videos converted to embedded references (portable)
labels.save("output.pkg.slp", embed="user")

# Selective: only embed specific frames, keep other videos as external paths
labels.save("output.pkg.slp", embed="user", embed_all_videos=False)

Trim labels and video

Extract a subset of frames with corresponding labels.

trim_video.py
import sleap_io as sio

# Load existing data
labels = sio.load_file("labels.slp")

# Create a new labels file with frames 1000-2000 from video 0
clip = labels.trim("clip.slp", list(range(1_000, 2_000)), video=0)

# The new file contains:
# - A trimmed video saved as "clip.mp4"
# - Labels with adjusted frame indices

Tip

The trim method automatically:

  • Creates a new video file with only the specified frames
  • Adjusts frame indices in the labels to match the new video
  • Preserves all instance data and tracks

See also

Labels.trim: Full trim method documentation

Make training/validation/test splits

Split your dataset for machine learning workflows.

make_splits.py
import sleap_io as sio

# Load source labels
labels = sio.load_file("labels.v001.slp")

# Make splits and export with embedded images
labels.make_training_splits(
    n_train=0.8,
    n_val=0.1,
    n_test=0.1,
    save_dir="split1",
    seed=42
)

# Splits are saved as self-contained SLP package files
labels_train = sio.load_file("split1/train.pkg.slp")
labels_val = sio.load_file("split1/val.pkg.slp")
labels_test = sio.load_file("split1/test.pkg.slp")

# Or get splits as a LabelsSet for programmatic access
labels_set = labels.make_training_splits(n_train=0.8, n_val=0.1, n_test=0.1)
train_labels = labels_set["train"]
val_labels = labels_set["val"]
test_labels = labels_set["test"]

Info

The .pkg.slp extension indicates a self-contained package with embedded images, making the splits portable and shareable.

See also

Video operations

Read video data

Load and access video frames directly.

read_video.py
import sleap_io as sio

video = sio.load_video("test.mp4")
n_frames, height, width, channels = video.shape

frame = video[0]  # Get first frame
height, width, channels = frame.shape

# Access specific frames
middle_frame = video[n_frames // 2]
last_frame = video[-1]

Info

Video backends are optional. Install the backend you need:

# imageio-ffmpeg is included by default

To check which backends are available or get installation help:

import sleap_io as sio
print(sio.get_available_video_backends())
print(sio.get_installation_instructions())

See also

Norpix .seq video

Load Norpix StreamPix .seq files the same way as any other video. The Norpix backend exposes per-frame timestamps and an auto-computed FPS derived from the timestamp stream, which is handy for high-speed behavioral recordings where the header FPS is often wrong.

read_seq.py
import sleap_io as sio

video = sio.load_video("recording.seq")
frame = video[0]

# Inspect backend-specific metadata
seq = video.backend
print(seq.fps)              # auto-computed from per-frame timestamps
print(seq.get_timestamps()[0])  # seconds-since-epoch of the first frame

Supported codecs: raw mono, raw RGB, JPEG, and PNG. The sio show CLI prints Norpix-specific header info (codec, bit depth, description, FPS), and sio reencode recording.seq -o recording.mp4 converts to MP4 via the Python path.

See also

Video backends — how SeqVideo fits alongside MediaVideo, HDF5Video, ImageVideo, and TiffVideo.

Re-encode video

Fix video seeking issues by re-encoding with optimal settings.

reencode_video.py
import sleap_io as sio

sio.save_video(sio.load_video("input.mp4"), "output.mp4")

Why re-encode?

Some video formats are not readily seekable at frame-level accuracy. Re-encoding with default settings ensures reliable seeking with minimal quality loss.

See also

save_video: Video saving options and codec settings

Virtual cropping and batch autocrop

Expose a virtual, on-read crop of a video — frames are decoded and sliced in memory, with no pixels copied or re-encoded (Video.crop / Video.from_crop). The crop is (x1, y1, x2, y2) in source pixels (x2/y2 exclusive); out-of-bounds regions are padded.

virtual_crop.py
import sleap_io as sio

full = sio.load_video("session.mp4")              # (1000, 1080, 1920, 3)
view = full.crop((320, 200, 576, 456))            # virtual view, no decode yet
view.shape                                         # (1000, 256, 256, 3)
view.is_cropped, view.crop_rect                    # True, (320, 200, 576, 456)
view.source_video is full                          # True - provenance preserved
frame = view[0]                                    # decode-then-slice (256, 256, 3)

# Other region specs: a bbox, an ROI (+ margin), or a fixed-size centered window.
view = full.crop(bbox=(320.0, 200.0, 576.0, 456.0))
view = full.crop(roi=my_shapely_poly, margin=8)
view = full.crop(center=(cx, cy), size=(128, 128))   # fixed shape; off-frame is padded

Batch autocrop (e.g. a multi-chamber rig). Apply a fixed set of per-chamber rects across many recordings and write one cropped file per (video x chamber). apply_crop bakes the virtual crop to disk and keeps source_video pointing at the uncropped original.

batch_autocrop.py
import sleap_io as sio
from pathlib import Path

# Chamber layout, defined once (x1, y1, x2, y2). 16-aligned dims avoid encoder padding.
chambers = {
    "A": (0, 0, 640, 480),
    "B": (640, 0, 1280, 480),
    "C": (0, 480, 640, 960),
    "D": (640, 480, 1280, 960),
}

out_dir = Path("crops")
out_dir.mkdir(exist_ok=True)
for path in Path("recordings").glob("*.mp4"):
    full = sio.load_video(path.as_posix())
    for name, rect in chambers.items():
        crop = sio.Video.from_crop(full, rect)
        crop.apply_crop((out_dir / f"{path.stem}_{name}.mp4").as_posix())

Prefer to stay lazy (no re-encode) and carry the crops in a labels file? Build the views into a Labels, save (crops ride a /video_crops dataset; pixels are untouched), and bake them all later in one call with Labels.apply_crops:

virtual_crop_slp.py
import sleap_io as sio

full = sio.load_video("session.mp4")
tiles = [sio.Video.from_crop(full, rect) for rect in chambers.values()]
sio.save_file(sio.Labels(videos=tiles), "session.slp")   # virtual; no re-encode

# Later - materialize every virtual crop to real files and update references:
sio.load_file("session.slp").apply_crops(video_dir="crops/")

The same step is available from the command line for an SLP that already carries virtual crops:

sio apply-crops session.slp -o baked.slp --video-dir crops/

See also

Switch video and image backends

Control which backend is used for video reading and embedded frame encoding.

Video reading backends

Choose which backend to use when loading videos with sio.load_video().

video_backends.py
import sleap_io as sio

# Set default video backend for reading video files
sio.set_default_video_plugin("opencv")
video = sio.load_video("test.mp4")

# Or use imageio-ffmpeg (bundled, always available)
sio.set_default_video_plugin("FFMPEG")
video = sio.load_video("test.mp4")

# Check current default
print(sio.get_default_video_plugin())  # "FFMPEG"

Video backend trade-offs

Backend Install Speed Notes
FFMPEG (FFMPEG) Bundled (always available) Moderate Default. Most reliable, best seeking accuracy
OpenCV (opencv) pip install sleap-io[opencv] Fastest May have platform-specific issues
PyAV (pyav) pip install sleap-io[pyav] Fast Alternative FFMPEG wrapper

Choose which backend to use when encoding frames in .pkg.slp files with sio.save_slp().

image_backends.py
import sleap_io as sio

# Load labels
labels = sio.load_slp("labels.slp")

# Set default image encoding backend
sio.set_default_image_plugin("opencv")

# Save with embedded frames using OpenCV encoding
labels.save("labels.pkg.slp", embed="all")

# Or specify plugin directly in save call
labels.save("labels.pkg.slp", embed="all", plugin="imageio")

# Check current default
print(sio.get_default_image_plugin())  # "opencv"

Automatic RGB/BGR conversion

When loading .pkg.slp files, sleap-io automatically handles RGB/BGR channel order conversions between different encoding and decoding backends. Frames will always load in RGB order regardless of which plugin was used for encoding vs decoding.

Image backend options

Backend Install Notes
imageio (imageio) Bundled (always available) Default. Encodes in RGB channel order
OpenCV (opencv) pip install sleap-io[opencv] Faster encoding. Encodes in BGR internally

RGB/BGR conversion is handled automatically — frames always load in RGB regardless of which backend was used for encoding.

Plugin vs backend terminology

  • Video plugins: Used by sio.load_video() for reading video files (opencv, FFMPEG, pyav)
  • Image plugins: Used by sio.save_slp() for encoding embedded frames in .pkg.slp files (opencv, imageio)
  • Both can be set via set_default_*_plugin() functions

See also

Segmentation

Work with dense per-pixel instance segmentation data (e.g., from Cellpose or StarDist).

Full reference

For the complete SegmentationMask / LabelImage API — factory methods, streaming writes, TIFF I/O, merging, and lazy loading — see the Segmentation reference page.

Import segmentation masks to SLP

Convert a (T, H, W) integer mask array into an SLP file with object metadata.

segmentation_to_slp.py
import numpy as np
import sleap_io as sio

# Load masks from segmentation tool output — (n_frames, height, width) int32
# 0 = background, 1+ = object IDs per frame
masks = np.load("cellpose_masks.npy")  # e.g., shape (100, 512, 512)

# Load the source video
video = sio.load_video("microscopy.tif")  # shape: (n_frames, height, width, channels)

# Convert to PredictedLabelImages with shared tracks across frames
label_images = sio.PredictedLabelImage.from_stack(
    masks, video=video, source="cellpose:nuclei",
    create_tracks=True, score=1.0,
)

# Build Labels — label_images are distributed to LabeledFrames automatically
labels = sio.Labels(label_images=label_images, videos=[video])
labels.save("segmentation.slp")
Custom tracks and categories per frame

For more control, use from_numpy() per frame with explicit track/category mappings:

tracks = {1: sio.Track(name="cell_A"), 2: sio.Track(name="cell_B")}
li = sio.PredictedLabelImage.from_numpy(
    masks[0], video=video, frame_idx=0,
    tracks=tracks, categories={1: "neuron", 2: "glia"},
    source="cellpose:nuclei", score=1.0,
)

How track sharing works

from_stack(create_tracks=True) maps each unique integer label to a Track object shared across all frames. If cell 5 appears in frames 0, 3, and 7, it gets the same Track in all three. This is useful when IDs are consistent across frames (e.g., after tracking). Without create_tracks, objects have no cross-frame identity.

See also

Import per-object binary masks to SLP

Convert per-object binary masks from tools like SAM or Mask R-CNN into an SLP file. Unlike from_stack (which takes a pre-composited integer array), from_binary_masks takes individual boolean masks per object.

binary_masks_to_slp.py
import numpy as np
import sleap_io as sio

# Per-object binary masks from SAM — (n_objects, height, width) bool array
# Each mask[i] is a binary mask for one detected object in a single frame
sam_masks = np.load("sam_masks.npy")  # e.g., shape (5, 512, 512) = 5 objects
object_scores = [0.95, 0.92, 0.88, 0.85, 0.80]

# Load the source video (e.g., a single-frame or multi-frame TIFF)
video = sio.load_video("microscopy.tif")  # shape: (n_frames, height, width, channels)

# Create a PredictedLabelImage with tracks and per-object scores
li = sio.PredictedLabelImage.from_binary_masks(
    sam_masks,
    create_tracks=True,          # auto-create one Track per mask
    scores=object_scores,        # per-object confidence → Info.score
    score=0.9,                   # image-level confidence
    source="sam",
)

# Add the label image to a Labels dataset via a LabeledFrame
lf = sio.LabeledFrame(video=video, frame_idx=0)
lf.append(li)
labels = sio.Labels(labeled_frames=[lf])
labels.save("sam_output.slp")

When to use which import method

  • from_binary_masks: You have per-object boolean masks for a single frame — shape (n_objects, H, W) (SAM, Mask R-CNN).
  • from_stack: You have a (n_frames, H, W) integer array where each pixel value is an object ID (Cellpose, StarDist).
  • from_numpy: You have a single (H, W) integer array for one frame.

See also

Stream large segmentation results to SLP

For datasets too large to hold in memory, write frames one at a time with constant memory usage.

streaming_segmentation.py
import numpy as np
import sleap_io as sio

# Load source video for frame data and metadata
video = sio.load_video("microscopy.tif")  # shape: (n_frames, height, width, channels)

# Stream frames to SLP — file is created lazily on first add()
# video is optional: associates all label images with this video in the SLP file
with sio.LabelImageWriter("output.slp", video=video) as writer:
    for frame_idx in range(len(video)):
        # Read the frame and run your segmentation model
        mask = run_segmentation(video[frame_idx])  # returns (H, W) int32

        li = sio.PredictedLabelImage.from_numpy(
            mask,
            source="cellpose:nuclei", score=1.0,
        )
        writer.add(li)
# File finalized and closed on context exit

Tracking across frames

By default, each frame's objects are independent — no cross-frame identity. To link objects across frames, pass a shared dict as tracks with create_tracks=True:

shared_tracks = {}  # accumulates {label_id: Track} across frames

li = sio.PredictedLabelImage.from_numpy(
    mask,
    tracks=shared_tracks, create_tracks=True,  # reuse existing, create new
    source="cellpose:nuclei", score=1.0,
)

Existing entries in the dict are reused and new label IDs get fresh Track objects added in place. This gives cross-frame identity without requiring all data in memory. The writer auto-collects new tracks from each add() call.

Memory and performance

The writer uses the chunked HDF5 format with gzip compression. Only one frame's compressed data is in memory at a time. A 42-frame (592x608) dataset compresses from ~60 MB raw to ~0.6 MB in the SLP file.

See also

Extract segmentation data from SLP

Load segmentation data and convert back to numpy arrays, TIFF files, or individual masks.

extract_segmentation.py
import numpy as np
import sleap_io as sio

# Load — pixel data is lazy (metadata queries don't decompress)
labels = sio.load_slp("segmentation.slp")

# Access label images through frames (annotations are nested in LabeledFrames)
lf = labels[0]  # first labeled frame
print(lf.frame_idx, len(lf.label_images))  # no decompression yet

# Inspect a single label image
li = lf.label_images[0]
print(li.n_objects, li.tracks, li.categories)  # still no decompression

# Extract all frames as (n_frames, height, width) numpy array
# .data triggers lazy decompression for each frame
all_label_images = labels.label_images  # flattened view across all frames
all_masks = np.stack([li.data for li in all_label_images])

# Export as TIFF stack (with sidecar metadata JSON)
sio.save_label_images("masks.tif", all_label_images, stack=True)

# Decompose one frame into per-object binary SegmentationMasks
for mask in li.to_masks():
    print(f"{mask.category}: {mask.area} pixels")
Expected output shapes

For a dataset with 42 frames at 592x608 with 21 objects:

  • all_masks.shape: (42, 592, 608), dtype int32
  • li.to_masks(): list of 21 SegmentationMask objects
  • Each mask: boolean array (592, 608) for one object

See also

Merge segmentation results

Combine label images from multiple SLP files into one (e.g., after parallel batch processing). This uses merge_label_images(), a specialized function that copies compressed HDF5 chunks directly between files without decompressing pixel data — much faster than loading everything into memory with Labels.merge().

merge_segmentation.py
import sleap_io as sio

# Merge batch results at the file level (no decompression needed)
merged = sio.merge_label_images(
    ["batch_0.slp", "batch_1.slp", "batch_2.slp"],
    "all_frames.slp",
)
print(f"Merged: {len(merged.label_images)} total frames")

Why not Labels.merge()?

Labels.merge() loads all data into Python objects, which is necessary for matching and resolving conflicts between keypoint annotations. For label images — which are typically non-overlapping frame batches from parallel segmentation — merge_label_images() skips all of that and concatenates the compressed pixel chunks directly. This makes it I/O-bound rather than CPU-bound.

See also

Parallel segmentation pipeline

Combine the streaming writer and the zero-decompression merger to build a parallel batch segmentation pipeline. Each worker streams its own shard with LabelImageWriter, and a final merge_label_images call concatenates the shards without decompressing a single pixel.

parallel_segmentation.py
import sleap_io as sio

def run_shard(video_path: str, frame_range: range, shard_path: str) -> None:
    """Segment a frame range and stream the results to one SLP shard."""
    video = sio.load_video(video_path)
    with sio.LabelImageWriter(shard_path, video=video) as writer:
        for frame_idx in frame_range:
            mask = run_segmentation(video[frame_idx])   # (H, W) int32
            li = sio.PredictedLabelImage.from_numpy(
                mask, source="cellpose:nuclei", score=1.0,
            )
            writer.add(li)

# Launch one worker per shard (e.g., with multiprocessing, joblib, or a job
# scheduler) so every worker streams to its own file.
run_shard("microscopy.tif", range(0, 500), "shard_0.slp")
run_shard("microscopy.tif", range(500, 1000), "shard_1.slp")
run_shard("microscopy.tif", range(1000, 1500), "shard_2.slp")

# Concatenate the shards into the final dataset in O(shards), without
# decompressing any pixel chunks.
merged = sio.merge_label_images(
    ["shard_0.slp", "shard_1.slp", "shard_2.slp"],
    "all_frames.slp",
)
print(f"Merged: {len(merged.label_images)} frames")

Why this is fast

Workers run in parallel and produce contiguous SLP shards that each hold only one frame's worth of compressed data in memory at a time. merge_label_images then copies compressed HDF5 chunks byte-for-byte between files — no decompression, no conflict resolution, no Python-level object reconstruction.

Rendering

Create videos and images with pose overlays for visualization and publication.

render_poses.py
import sleap_io as sio

labels = sio.load_slp("predictions.slp")

# Render full video (MP4 with skeleton overlays at source resolution)
labels.render("output.mp4")

# Fast preview for iteration (0.25x resolution, faster encoding)
labels.render("preview.mp4", preset="preview")

# Single frame to PNG image
sio.render_image(labels[0], "frame.png")
CLI
sio render -i predictions.slp -o output.mp4
sio render -i predictions.slp --preset preview
sio render -i predictions.slp --lf 0  # Single frame

Motion trails

Draw fading trajectory trails behind each instance to visualize movement over time:

render_trails.py
import sleap_io as sio

labels = sio.load_slp("predictions.slp")

# Trails follow track identities, so the data should be tracked.
sio.render_video(
    labels,
    "trails.mp4",
    show_trails=True,
    trail_length=10,      # frames of history per trail
    trail_node="centroid",  # or a node name / list of node names
)

Trails need tracks

Trails are drawn per track, so untracked data produces little or no trail. Run tracking first, or merge with track="name" if your tracks are named. See Rendering → Motion trails for all trail_* options.

Segmentation overlay rendering

Render pose predictions on top of a segmentation mask (or render masks on bare images, no labels file needed). The overlay pipeline accepts 2-D label images, 3-D label stacks, or a directory of per-frame TIFFs.

segmentation_overlay.py
import numpy as np
import sleap_io as sio

labels = sio.load_slp("predictions.slp")
lf = labels[0]

# Overlay a (H, W) integer label image on a single labeled frame
label_mask = np.load("frame0_mask.npy")  # (H, W) int
img = sio.render_image(
    lf,
    overlay=label_mask,
    overlay_alpha=0.4,
    overlay_outline=True,
)

# Overlay-only mode (no poses) — renders masks on an arbitrary image
frame = sio.load_video("clip.mp4")[0]
masked = sio.render_image(image=frame, overlay=label_mask)

# Video rendering with a 3-D label stack (one mask per frame)
stack = np.load("masks_stack.npy")  # (T, H, W) int
sio.render_video(labels, "overlay.mp4", overlay=stack)

# Standalone draw_* helpers compose overlays manually onto any image
canvas = frame.copy()
sio.draw_label_image(canvas, label_mask, alpha=0.4, outline=True)
sio.draw_bboxes(canvas, lf.bboxes)
sio.draw_masks(canvas, [m for m in lf.masks])
sio.draw_rois(canvas, lf.rois, fill_alpha=0.3)
CLI
# Overlay segmentation on rendered poses
sio render predictions.slp --overlay masks.tif --overlay-alpha 0.4

# Overlay-only mode: no labels file needed
sio render --images frames/ --overlay masks.tif -o output.mp4

See also

See the Rendering Guide for complete documentation including color schemes, marker shapes, custom callbacks, and all CLI options.