Merge pull request #1064 from luxonis/develop

DepthAI SDK 1.12.0

Author: daniilpastukhov

depthai_sdk/docs/source/conf.py

@@ -22,7 +22,7 @@
 author = 'Luxonis'
 
 # The full version, including alpha/beta/rc tags
-release = '1.11.0'
+release = '1.12.0'
 
 
 # -- General configuration ---------------------------------------------------

depthai_sdk/examples/CameraComponent/camera_control_with_nn.py

@@ -0,0 +1,10 @@
+from depthai_sdk import OakCamera
+
+with OakCamera() as oak:
+    color = oak.create_camera('color')
+    face_det = oak.create_nn('face-detection-retail-0004', color)
+    # Control the camera's exposure/focus based on the (largest) detected face
+    color.control_with_nn(face_det, auto_focus=True, auto_exposure=True, debug=False)
+
+    oak.visualize(face_det, fps=True)
+    oak.start(blocking=True)

depthai_sdk/examples/StereoComponent/stereo_encoded.py

@@ -6,7 +6,7 @@
     stereo = oak.create_stereo('800p', fps=30, encode='h264')
 
     # Set on-device output colorization, works only for encoded output
-    stereo.set_colormap(dai.Colormap.STEREO_JET)
+    stereo.set_colormap(dai.Colormap.JET)
 
     oak.visualize(stereo.out.encoded, fps=True)
     oak.start(blocking=True)

depthai_sdk/examples/trigger_action/custom_trigger.py

@@ -27,7 +27,7 @@ def my_condition(packet) -> bool:
 with OakCamera() as oak:
     color = oak.create_camera('color', fps=30)
     stereo = oak.create_stereo('800p')
-    stereo.config_stereo(align='color')
+    stereo.config_stereo(align=color)
 
     trigger = Trigger(input=stereo.out.depth, condition=my_condition, cooldown=30)
     action = RecordAction(

depthai_sdk/requirements.txt

@@ -4,7 +4,7 @@ opencv-contrib-python>4
 blobconverter>=1.4.1
 pytube>=12.1.0
 --extra-index-url https://artifacts.luxonis.com/artifactory/luxonis-python-snapshot-local/
-depthai==2.21.2
+depthai==2.22.0
 PyTurboJPEG==1.6.4
 marshmallow==3.17.0
 xmltodict

depthai_sdk/sdk_tests/components/nn/test_nn_component.py

@@ -91,3 +91,15 @@ def callback(packet):
             if not oak_camera.poll():
                 raise RuntimeError('Polling failed')
             time.sleep(0.1)
+
+
+def test_encoded_output():
+    with OakCamera() as oak_camera:
+        camera = oak_camera.create_camera('color', '1080p', encode='h264')
+
+        oak_camera.callback(camera.out.encoded, lambda x: print(x))
+        oak_camera.start(blocking=False)
+
+        for i in range(10):
+            oak_camera.poll()
+            time.sleep(0.1)

depthai_sdk/sdk_tests/components/stereo/test_stereo_component.py

@@ -1,19 +1,21 @@
 import time
 
+import depthai as dai
 import pytest
+
 from depthai_sdk.oak_camera import OakCamera
-import depthai as dai
 
 
 def test_stereo_output():
     with OakCamera() as oak_camera:
         if dai.CameraBoardSocket.LEFT not in oak_camera.sensors:
             pytest.skip('Looks like camera does not have mono pair, skipping...')
         else:
-            stereo = oak_camera.create_stereo('400p')
+            stereo = oak_camera.create_stereo('800p', encode='h264')
 
             oak_camera.callback([stereo.out.depth, stereo.out.disparity,
-                                 stereo.out.rectified_left, stereo.out.rectified_right], callback=lambda x: None)
+                                 stereo.out.rectified_left, stereo.out.rectified_right,
+                                 stereo.out.encoded], callback=lambda x: None)
             oak_camera.start(blocking=False)
 
             for i in range(10):

depthai_sdk/sdk_tests/test_examples.py

@@ -1,27 +1,38 @@
+import os
 import subprocess
 import sys
 import time
 from pathlib import Path
 
 import cv2
+import pytest
 
-EXAMPLES_DIR = Path(__file__).parent.parent.parent / "examples"
+EXAMPLES_DIR = Path(__file__).parents[1] / 'examples'
 
+# Create a temporary directory for the tests
+Path('/tmp/depthai_sdk_tests').mkdir(exist_ok=True)
+os.chdir('/tmp/depthai_sdk_tests')
 
-def test_examples():
-    python_executable = Path(sys.executable)
-    for example in EXAMPLES_DIR.rglob("**/*.py"):
-        print(f"Running example: {example.name}")
-
-        result = subprocess.Popen(f"{python_executable} {example}", stdout=subprocess.PIPE, stderr=subprocess.PIPE,
-                                  env={"DISPLAY": ""}, shell=True)
-
-        time.sleep(5)
-        result.kill()
-        time.sleep(5)
-        print('Stderr: ', result.stderr.read().decode())
 
-        # if result.returncode and result.returncode != 0:
-        #     assert False, f"{example} raised an exception: {result.stderr}"
-
-        cv2.destroyAllWindows()
+@pytest.mark.parametrize('example', list(EXAMPLES_DIR.rglob("**/*.py")))
+def test_examples(example):
+    print(f"Running {example}")
+    python_executable = Path(sys.executable)
+    result = subprocess.Popen(f"{python_executable} {example}",
+                              stdout=subprocess.PIPE,
+                              stderr=subprocess.PIPE,
+                              env={
+                                  'DISPLAY': '',
+                                  'PYTHONPATH': f'{os.environ["PYTHONPATH"]}:{EXAMPLES_DIR.parent}'
+                              },
+                              shell=True)
+
+    time.sleep(5)
+    result.kill()
+    time.sleep(5)
+    print('Stderr: ', result.stderr.read().decode())
+
+    if result.returncode and result.returncode != 0:
+        assert False, f"{example} raised an exception: {result.stderr}"
+
+    cv2.destroyAllWindows()

depthai_sdk/setup.py

@@ -9,7 +9,7 @@
 
 setup(
     name='depthai-sdk',
-    version='1.11.0',
+    version='1.12.0',
     description='This package provides an abstraction of the DepthAI API library.',
     long_description=io.open("README.md", encoding="utf-8").read(),
     long_description_content_type="text/markdown",
@@ -30,7 +30,8 @@
         "replay": ['mcap>=0.0.10',
                    'mcap-ros1-support==0.0.8',
                    'rosbags==0.9.11'],
-        "record": ['av']
+        "record": ['av'],
+        "test": ['pytest']
     },
     project_urls={
         "Bug Tracker": "https://github.com/luxonis/depthai/issues",

depthai_sdk/src/depthai_sdk/__init__.py

@@ -1,5 +1,6 @@
 from depthai_sdk.args_parser import ArgsParser
 from depthai_sdk.classes.enum import ResizeMode
+from depthai_sdk.constants import CV2_HAS_GUI_SUPPORT
 from depthai_sdk.logger import set_logging_level
 from depthai_sdk.oak_camera import OakCamera
 from depthai_sdk.oak_device import OakDevice
@@ -10,7 +11,7 @@
 from depthai_sdk.utils import _create_config, get_config_field
 from depthai_sdk.visualize import *
 
-__version__ = '1.11.0'
+__version__ = '1.12.0'
 
 
 def __import_sentry(sentry_dsn: str) -> None: