Merge pull request #921 from luxonis/sdk/optimization

Release 1.9.3

Author: daniilpastukhov

depthai_sdk/MANIFEST.in

@@ -7,6 +7,7 @@ recursive-include src/depthai_sdk/nn_models *.yml
 recursive-include src/depthai_sdk/nn_models config.json
 
 include src/depthai_sdk/oak_outputs/*.py
+include src/depthai_sdk/oak_outputs/xout/*.py
 include src/depthai_sdk/readers/*.py
 recursive-include src/depthai_sdk/recorders *.py
 include src/depthai_sdk/visualize/*.py

depthai_sdk/docs/source/components/camera_component.rst

@@ -10,6 +10,8 @@ Usage
 
 .. code-block:: python
 
+    from depthai_sdk import OakCamera
+
     with OakCamera() as oak:
         # Create color camera
         color = oak.create_camera('color')
@@ -30,6 +32,6 @@ Component outputs
 Reference
 #########
 
-.. autoclass:: depthai_sdk.CameraComponent
+.. autoclass:: depthai_sdk.components.CameraComponent
     :members:
     :undoc-members:

depthai_sdk/docs/source/components/imu_component.rst

@@ -8,6 +8,9 @@ Usage
 
 .. code-block:: python
 
+    from depthai_sdk import OakCamera
+    from depthai_sdk.classes import IMUPacket
+
     with OakCamera() as oak:
         imu = oak.create_imu()
         imu.config_imu(report_rate=400, batch_report_threshold=5)
@@ -26,6 +29,6 @@ Component outputs
 Reference
 #########
 
-.. autoclass:: depthai_sdk.IMUComponent
+.. autoclass:: depthai_sdk.components.IMUComponent
     :members:
     :undoc-members:

depthai_sdk/docs/source/components/nn_component.rst

@@ -57,8 +57,8 @@ Example usage:
     import numpy as np
     from depthai import NNData
 
-    from depthai_sdk import OakCamera, Detections
-
+    from depthai_sdk import OakCamera
+    from depthai_sdk.classes import Detections
 
     def decode(nn_data: NNData):
         layer = nn_data.getFirstLayerFp16()
@@ -89,7 +89,7 @@ Example usage:
 Reference
 #########
 
-.. autoclass:: depthai_sdk.NNComponent
+.. autoclass:: depthai_sdk.components.NNComponent
     :members:
     :undoc-members:
 

depthai_sdk/docs/source/components/stereo_component.rst

@@ -9,6 +9,8 @@ Usage
 
 .. code-block:: python
 
+    from depthai_sdk import OakCamera
+
     with OakCamera() as oak:
         # Create stereo component, initialize left/right MonoCamera nodes for 800P and 60FPS
         stereo = oak.create_stereo('800p', fps=60)
@@ -28,6 +30,6 @@ Component outputs
 Reference
 #########
 
-.. autoclass:: depthai_sdk.StereoComponent
+.. autoclass:: depthai_sdk.components.StereoComponent
     :members:
     :undoc-members:

depthai_sdk/docs/source/features/components.rst

@@ -13,7 +13,7 @@ their initialization, configuration, and linking. This improves ease of use when
 
 **Reference**
 
-.. autoclass:: depthai_sdk.Component
+.. autoclass:: depthai_sdk.components.Component
     :members:
     :undoc-members:
 

depthai_sdk/docs/source/features/packets.rst

@@ -19,14 +19,15 @@ Example
 .. code-block:: python
 
     import cv2
-    from depthai_sdk import OakCamera, DetectionPacket, Visualizer
+    from depthai_sdk import OakCamera
+    from depthai_sdk.classes import DetectionPacket
 
     with OakCamera() as oak:
         color = oak.create_camera('color')
         nn = oak.create_nn('mobilenet-ssd', color)
 
         # Callback
-        def cb(packet: DetectionPacket, visualizer: Visualizer):
+        def cb(packet: DetectionPacket):
             print(packet.img_detections)
             cv2.imshow(packet.name, packet.frame)
 

depthai_sdk/examples/CameraComponent/mono_400p.py

@@ -1,10 +1,7 @@
-import cv2
 from depthai_sdk import OakCamera
 
 with OakCamera() as oak:
     left = oak.create_camera('left', resolution='400p', fps=60)
     right = oak.create_camera('right', resolution='400p', fps=60)
-
     oak.visualize([left, right], fps=True)
-
     oak.start(blocking=True)

depthai_sdk/examples/NNComponent/age-gender.py

@@ -1,10 +1,13 @@
 import cv2
 import numpy as np
 
-from depthai_sdk import OakCamera, TextPosition, Visualizer, TwoStagePacket
+from depthai_sdk import OakCamera
+from depthai_sdk.classes import TwoStagePacket
+from depthai_sdk.visualize.configs import TextPosition
 
 
-def callback(packet: TwoStagePacket, visualizer: Visualizer):
+def callback(packet: TwoStagePacket):
+    visualizer = packet.visualizer
     for det, rec in zip(packet.detections, packet.nnData):
         age = int(float(np.squeeze(np.array(rec.getLayerFp16('age_conv3')))) * 100)
         gender = np.squeeze(np.array(rec.getLayerFp16('prob')))
@@ -20,7 +23,6 @@ def callback(packet: TwoStagePacket, visualizer: Visualizer):
 
 with OakCamera() as oak:
     color = oak.create_camera('color')
-
     det = oak.create_nn('face-detection-retail-0004', color)
     det.config_nn(resize_mode='crop')
 

depthai_sdk/examples/NNComponent/custom_decode.py

@@ -3,7 +3,8 @@
 import numpy as np
 from depthai import NNData
 
-from depthai_sdk import OakCamera, Detections, Visualizer, DetectionPacket
+from depthai_sdk import OakCamera
+from depthai_sdk.classes import Detections, DetectionPacket
 
 
 def decode(nn_data: NNData) -> Detections:
@@ -21,7 +22,8 @@ def decode(nn_data: NNData) -> Detections:
     return dets
 
 
-def callback(packet: DetectionPacket, visualizer: Visualizer):
+def callback(packet: DetectionPacket):
+    visualizer = packet.visualizer
     frame = packet.frame
     frame = visualizer.draw(frame)
     cv2.imshow('Custom decode function', frame)