sinapsis-supervision 0.1.2

Templates for multi-object tracking using the ByteTrack algorithm

Sinapsis Supervision

Templates for multi-object tracking using the ByteTrack model

🐍 Installation • 🚀 Features • 📙 Documentation • 🔍 License

The sinapsis-supervision module offers a powerful and flexible implementation for multi-object tracking using the ByteTrack algorithm. It allows users to easily configure and run tracking pipelines for video input processing, object detection, and tracking tasks.

🐍 Installation

Install using your package manager of choice. We encourage the use of uv

Example with uv:

  uv pip install sinapsis-supervision --extra-index-url https://pypi.sinapsis.tech

or with raw pip:

  pip install sinapsis-supervision --extra-index-url https://pypi.sinapsis.tech

🚀 Features

Templates Supported

The Sinapsis ByteTrack module provides a template for multi-object tracking using the ByteTrack algorithm. Currently, the package includes the following template:

• ByteTrack: A template for tracking objects across video frames, with customizable parameters for track activation, matching thresholds, and occlusion handling.

[!TIP] Use CLI command sinapsis info --all-template-names to show a list with all the available Template names installed with Sinapsis Supervision.

[!TIP] Use CLI command sinapsis info --example-template-config TEMPLATE_NAME to produce an example Agent config for the Template specified in TEMPLATE_NAME.

For example, for ByteTrack use sinapsis info --example-template-config ByteTrack to produce the following example config:

agent:
  name: my_test_agent
templates:
- template_name: InputTemplate
  class_name: InputTemplate
  attributes: {}
- template_name: ByteTrack
  class_name: ByteTrack
  template_input: InputTemplate
  attributes:
    track_activation_threshold: 0.2
    lost_track_buffer: 30
    minimum_matching_threshold: 0.7
    frame_rate: 30
    minimum_consecutive_frames: 1

📚 Example Usage

Below is an example YAML configuration for processing a video file and visualizing tracking results using the Sinapsis Supervision template. This configuration loads a video with the VideoReaderCV2, performs object detection and real-time tracking with the UltralyticsPredict and ByteTrack templates, draws bounding boxes around detected objects with the BBoxDrawer, and saves the output as a new video file using the VideoWriterCV2.

Config

agent:
  name: cotracker_agent

templates:
  - template_name: InputTemplate
    class_name: InputTemplate
    attributes: {}

  - template_name : VideoReaderCV2
    class_name: VideoReaderCV2
    template_input: InputTemplate
    attributes:
      video_file_path : "artifacts/palace.mp4"
      batch_size: 16

  - template_name: CoTrackerOnline
    class_name: CoTrackerOnline
    template_input: VideoReaderCV2
    attributes:
      model_variant: baseline
      device: cuda
      grid_size: 15

  - template_name: CoTrackerVisualizer
    class_name: CoTrackerVisualizer
    template_input: CoTrackerOnline
    attributes:
      device : cuda
      linewidth: 3
      overwrite: true

  - template_name: VideoWriterCV2
    class_name: VideoWriterCV2
    template_input: CoTrackerVisualizer
    attributes:
      destination_path: "artifacts/result.mp4"
      height: -1
      width: -1
      fps: 30

This configuration defines an agent and a sequence of templates for video processing, object detection, real-time tracking, and visualization.

IMPORTANT: The VideoReaderCV2, BBoxDrawer, and VideoWriterCV2 templates are part of the sinapsis-data-readers, sinapsis-data-visualization, and sinapsis-data-writers packages, respectively. To use this example, ensure that you have installed these packages.

To run the config, use the CLI:

sinapsis run name_of_config.yml

📙 Documentation

Documentation for this and other sinapsis packages is available on the sinapsis website

Tutorials for different projects within sinapsis are available at sinapsis tutorials page

🔍 License

This project is licensed under the AGPLv3 license, which encourages open collaboration and sharing. For more details, please refer to the LICENSE file.

For commercial use, please refer to our official Sinapsis website for information on obtaining a commercial license.