DepthAI SDK

DepthAI SDK is a Python package built on top of the depthai-python API library that improves ease of use when developing apps for OAK devices.

https://user-images.githubusercontent.com/18037362/142909472-eae7bed1-695b-48ec-8895-76982989de11.png

OakCamera class

The OakCamera class abstracts DepthAI API pipeline building, different camera permutations, stream recording/replaying, adds debugging features, handles AI model sourcing and decoding, does message syncing & visualization, and much more.

Script below demonstrates how you can easily accomplish complex tasks, that would otherwise take 100s of lines of code, and a few hours of assembling code pieces together.

 1from depthai_sdk import OakCamera, ResizeMode
 2
 3# Download a public depthai-recording and replay it
 4with OakCamera(replay='cars-tracking-above-01') as oak:
 5   # Create color camera
 6   color = oak.create_camera('color')
 7
 8   # Download & run pretrained vehicle detection model and track detections
 9   nn = oak.create_nn('vehicle-detection-0202', color, tracker=True)
10   nn.config_nn(resize_mode=ResizeMode.STRETCH)
11
12   # Visualize tracklets, show FPS
13   visualizer = oak.visualize(nn.out.tracker, fps=True)
14   visualizer.tracking(line_thickness=5).text(auto_scale=True)
15
16   # Start the app in blocking mode
17   oak.start(blocking=True)
https://user-images.githubusercontent.com/18037362/197657107-1c0a73ae-9022-4a85-abe2-892725edaa5f.gif

A public video gets downloaded and frames are sent to the OAK camera. On the OAK camera, we run vehicle-detection-0202, which is an SDK supported model. Afterwards, we use object tracker for tracking these detections over time. We visualize tracking results and configure visualizer to best fit our needs.

https://user-images.githubusercontent.com/18037362/197664572-33d8dc9e-dd35-4e73-8291-30bb8ec641d5.png

Car tracking pipeline from oak.show_graph()

Note

This class will be in alpha stage until depthai-sdk 2.0.0, so there will likely be some API changes.

Installation

To install this package, run the following command in your terminal window

python3 -m pip install depthai-sdk

Got questions?

We’re always happy to help with development or other questions you might have.