Merge pull request #398 from luxonis/tests_fixes

Fix examples/tests

Author: SzabolcsGergely

depthai-core

@@ -87,97 +87,97 @@ endif()
 # Add examples
 
 ## Bootloader
-add_python_example(bootloader_config bootloader_config.py)
-add_python_example(bootloader_version bootloader_version.py)
-add_python_example(flash_bootloader flash_bootloader.py)
+# add_python_example(bootloader_config bootloader/bootloader_config.py)
+add_python_example(bootloader_version bootloader/bootloader_version.py)
+# add_python_example(flash_bootloader bootloader/flash_bootloader.py)
 
 ## Calibration
-add_python_example(calibration_flash_v5 calibration_flash_v5.py)
-add_python_example(calibration_flash calibration_flash.py)
-add_python_example(calibration_load calibration_load.py)
-add_python_example(calibration_reader calibration_reader.py)
+# add_python_example(calibration_flash_v5 calibration/calibration_flash_v5.py)
+# add_python_example(calibration_flash calibration/calibration_flash.py)
+add_python_example(calibration_load calibration/calibration_load.py)
+add_python_example(calibration_reader calibration/calibration_reader.py)
 
 ## ColorCamera
-add_python_example(autoexposure_roi autoexposure_roi.py)
-add_python_example(rgb_camera_control rgb_camera_control.py)
-add_python_example(rgb_preview rgb_preview.py)
-add_python_example(rgb_scene rgb_scene.py)
-add_python_example(rgb_video rgb_video.py)
+add_python_example(autoexposure_roi ColorCamera/autoexposure_roi.py)
+add_python_example(rgb_camera_control ColorCamera/rgb_camera_control.py)
+add_python_example(rgb_preview ColorCamera/rgb_preview.py)
+add_python_example(rgb_scene ColorCamera/rgb_scene.py)
+add_python_example(rgb_video ColorCamera/rgb_video.py)
 
 ## EdgeDetector
-add_python_example(edge_detector edge_detector.py)
+add_python_example(edge_detector EdgeDetector/edge_detector.py)
 
 ## FeatureTracker
-add_python_example(feature_tracker feature_tracker.py)
-add_python_example(feature_detector feature_detector.py)
+add_python_example(feature_tracker FeatureTracker/feature_tracker.py)
+add_python_example(feature_detector FeatureTracker/feature_detector.py)
 
 ## HostSide
-add_python_example(device_queue_event device_queue_event.py)
-add_python_example(opencv_support opencv_support.py)
-add_python_example(queue_add_callback queue_add_callback.py)
+add_python_example(device_queue_event host_side/device_queue_event.py)
+add_python_example(opencv_support host_side/opencv_support.py)
+add_python_example(queue_add_callback host_side/queue_add_callback.py)
 
 ## ImageManip
-add_python_example(image_manip_rotate image_manip_rotate.py)
-add_python_example(image_manip_tiling image_manip_tiling.py)
-add_python_example(rgb_rotate_warp rgb_rotate_warp.py)
+add_python_example(image_manip_rotate ImageManip/image_manip_rotate.py)
+add_python_example(image_manip_tiling ImageManip/image_manip_tiling.py)
+add_python_example(rgb_rotate_warp ImageManip/rgb_rotate_warp.py)
 
 ## IMU
-add_python_example(imu_gyroscope_accelerometer imu_gyroscope_accelerometer.py)
-add_python_example(imu_rotation_vector imu_rotation_vector.py)
+add_python_example(imu_gyroscope_accelerometer IMU/imu_gyroscope_accelerometer.py)
+add_python_example(imu_rotation_vector IMU/imu_rotation_vector.py)
 
 ## Mixed
-add_python_example(mono_depth_mobilenetssd mono_depth_mobilenetssd.py)
-add_python_example(rgb_encoding_mobilenet rgb_encoding_mobilenet.py)
-add_python_example(rgb_encoding_mono_mobilenet rgb_encoding_mono_mobilenet.py)
-add_python_example(rgb_encoding_mono_mobilenet_depth rgb_encoding_mono_mobilenet_depth.py)
+add_python_example(mono_depth_mobilenetssd mixed/mono_depth_mobilenetssd.py)
+add_python_example(rgb_encoding_mobilenet mixed/rgb_encoding_mobilenet.py)
+add_python_example(rgb_encoding_mono_mobilenet mixed/rgb_encoding_mono_mobilenet.py)
+add_python_example(rgb_encoding_mono_mobilenet_depth mixed/rgb_encoding_mono_mobilenet_depth.py)
 
 ## MobileNet
-add_python_example(mono_mobilenet mono_mobilenet.py)
-add_python_example(rgb_mobilenet_4k rgb_mobilenet_4k.py)
-add_python_example(rgb_mobilenet rgb_mobilenet.py)
-add_python_example(video_mobilenet video_mobilenet.py)
+add_python_example(mono_mobilenet MobileNet/mono_mobilenet.py)
+add_python_example(rgb_mobilenet_4k MobileNet/rgb_mobilenet_4k.py)
+add_python_example(rgb_mobilenet MobileNet/rgb_mobilenet.py)
+add_python_example(video_mobilenet MobileNet/video_mobilenet.py)
 
 ## MonoCamera
-add_python_example(mono_preview mono_preview.py)
-add_python_example(mono_full_resolution_saver mono_full_resolution_saver.py)
-add_python_example(mono_camera_control mono_camera_control.py)
+add_python_example(mono_preview MonoCamera/mono_preview.py)
+add_python_example(mono_full_resolution_saver MonoCamera/mono_full_resolution_saver.py)
+add_python_example(mono_camera_control MonoCamera/mono_camera_control.py)
 
 ## NeuralNetwork
-add_python_example(concat_multiple_input concat_multiple_input.py)
-add_python_example(normalization_multiple_input normalization_multiple_input.py)
+# add_python_example(concat_multiple_input NeuralNetwork/concat_multiple_input.py)
+# add_python_example(normalization_multiple_input NeuralNetwork/normalization_multiple_input.py)
 
 ## ObjectTracker
-add_python_example(object_tracker_video object_tracker_video.py)
-add_python_example(object_tracker object_tracker.py)
-add_python_example(spatial_object_tracker_mobilenet spatial_object_tracker_mobilenet.py)
-add_python_example(spatial_object_tracker_yolo spatial_object_tracker_yolo.py)
+add_python_example(object_tracker_video ObjectTracker/object_tracker_video.py)
+add_python_example(object_tracker ObjectTracker/object_tracker.py)
+add_python_example(spatial_object_tracker ObjectTracker/spatial_object_tracker.py)
+# add_python_example(spatial_object_tracker_yolo ObjectTracker/spatial_object_tracker_yolo.py)
 
 ## Script
-add_python_example(script_camera_control script_camera_control.py)
+add_python_example(script_camera_control Script/script_camera_control.py)
 
 ## SpatialDetection
-add_python_example(spatial_tiny_yolo_v3 spatial_tiny_yolo.py)
-add_python_example(spatial_tiny_yolo_v4 spatial_tiny_yolo.py)
-add_python_example(spatial_mobilenet spatial_mobilenet.py)
-add_python_example(spatial_mobilenet_mono spatial_mobilenet_mono.py)
-add_python_example(spatial_location_calculator spatial_location_calculator.py)
+add_python_example(spatial_tiny_yolo_v3 SpatialDetection/spatial_tiny_yolo.py yolo3)
+add_python_example(spatial_tiny_yolo_v4 SpatialDetection/spatial_tiny_yolo.py yolo4)
+add_python_example(spatial_mobilenet SpatialDetection/spatial_mobilenet.py)
+add_python_example(spatial_mobilenet_mono SpatialDetection/spatial_mobilenet_mono.py)
+add_python_example(spatial_location_calculator SpatialDetection/spatial_location_calculator.py)
 
 ## StereoDepth
-add_python_example(depth_crop_control depth_crop_control.py)
-add_python_example(depth_preview depth_preview.py)
-add_python_example(rgb_depth_aligned rgb_depth_aligned.py)
-add_python_example(stereo_depth_from_host stereo_depth_from_host.py)
-add_python_example(stereo_depth_video stereo_depth_video.py)
+add_python_example(depth_crop_control StereoDepth/depth_crop_control.py)
+add_python_example(depth_preview StereoDepth/depth_preview.py)
+add_python_example(rgb_depth_aligned StereoDepth/rgb_depth_aligned.py)
+add_python_example(stereo_depth_from_host StereoDepth/stereo_depth_from_host.py)
+add_python_example(stereo_depth_video StereoDepth/stereo_depth_video.py)
 
 ## SystemLogger
-add_python_example(system_information system_information.py)
+add_python_example(system_information SystemLogger/system_information.py)
 
 ## VideoEncoder
-add_python_example(encoding_max_limit encoding_max_limit.py)
-add_python_example(rgb_encoding rgb_encoding.py)
-add_python_example(rgb_full_resolution_saver rgb_full_resolution_saver.py)
-add_python_example(rgb_mono_encoding rgb_mono_encoding.py)
+add_python_example(encoding_max_limit VideoEncoder/encoding_max_limit.py)
+add_python_example(rgb_encoding VideoEncoder/rgb_encoding.py)
+add_python_example(rgb_full_resolution_saver VideoEncoder/rgb_full_resolution_saver.py)
+add_python_example(rgb_mono_encoding VideoEncoder/rgb_mono_encoding.py)
 
 ## Yolo
-add_python_example(tiny_yolo_v3_device_side_decoding tiny_yolo_v3_device_side_decoding.py)
-add_python_example(tiny_yolo_v4_device_side_decoding tiny_yolo_v4_device_side_decoding.py)
+add_python_example(tiny_yolo_v3_device_side_decoding Yolo/tiny_yolo.py yolo3)
+add_python_example(tiny_yolo_v4_device_side_decoding Yolo/tiny_yolo.py yolo4)

examples/CMakeLists.txt

@@ -24,28 +24,28 @@
 pipeline = dai.Pipeline()
 
 # Define source and outputs
-camRgb = pipeline.createColorCamera()
+camRgb = pipeline.create(dai.node.ColorCamera)
 camRgb.setPreviewSize(*previewSize)
 camRgb.setInterleaved(False)
 
-camControlIn = pipeline.createXLinkIn()
+camControlIn = pipeline.create(dai.node.XLinkIn)
 camControlIn.setStreamName('camControl')
 camControlIn.out.link(camRgb.inputControl)
 
 # Define a neural network that will make predictions based on the source frames
-nn = pipeline.createMobileNetDetectionNetwork()
+nn = pipeline.create(dai.node.MobileNetDetectionNetwork)
 nn.setConfidenceThreshold(0.5)
 nn.setBlobPath(nnPath)
 nn.setNumInferenceThreads(2)
 nn.input.setBlocking(False)
 camRgb.preview.link(nn.input)
 
 # Linking
-xoutRgb = pipeline.createXLinkOut()
+xoutRgb = pipeline.create(dai.node.XLinkOut)
 xoutRgb.setStreamName("rgb")
 camRgb.preview.link(xoutRgb.input)
 
-nnOut = pipeline.createXLinkOut()
+nnOut = pipeline.create(dai.node.XLinkOut)
 nnOut.setStreamName("nn")
 nn.out.link(nnOut.input)
 

examples/ColorCamera/autoexposure_roi.py

@@ -30,15 +30,15 @@ def clamp(num, v0, v1):
 pipeline = dai.Pipeline()
 
 # Define sources and outputs
-camRgb = pipeline.createColorCamera()
-videoEncoder = pipeline.createVideoEncoder()
-stillEncoder = pipeline.createVideoEncoder()
-
-controlIn = pipeline.createXLinkIn()
-configIn = pipeline.createXLinkIn()
-videoMjpegOut = pipeline.createXLinkOut()
-stillMjpegOut = pipeline.createXLinkOut()
-previewOut = pipeline.createXLinkOut()
+camRgb = pipeline.create(dai.node.ColorCamera)
+videoEncoder = pipeline.create(dai.node.VideoEncoder)
+stillEncoder = pipeline.create(dai.node.VideoEncoder)
+
+controlIn = pipeline.create(dai.node.XLinkIn)
+configIn = pipeline.create(dai.node.XLinkIn)
+videoMjpegOut = pipeline.create(dai.node.XLinkOut)
+stillMjpegOut = pipeline.create(dai.node.XLinkOut)
+previewOut = pipeline.create(dai.node.XLinkOut)
 
 controlIn.setStreamName('control')
 configIn.setStreamName('config')

examples/ColorCamera/rgb_camera_control.py

@@ -7,8 +7,8 @@
 pipeline = dai.Pipeline()
 
 # Define source and output
-camRgb = pipeline.createColorCamera()
-xoutRgb = pipeline.createXLinkOut()
+camRgb = pipeline.create(dai.node.ColorCamera)
+xoutRgb = pipeline.create(dai.node.XLinkOut)
 
 xoutRgb.setStreamName("rgb")
 

examples/ColorCamera/rgb_preview.py

@@ -10,14 +10,14 @@
 # Create pipeline
 pipeline = dai.Pipeline()
 
-camRgb = pipeline.createColorCamera()
+camRgb = pipeline.create(dai.node.ColorCamera)
 camRgb.setIspScale(1,3)
 
-xoutRgb = pipeline.createXLinkOut()
+xoutRgb = pipeline.create(dai.node.XLinkOut)
 xoutRgb.setStreamName("video")
 camRgb.video.link(xoutRgb.input)
 
-camControlIn = pipeline.createXLinkIn()
+camControlIn = pipeline.create(dai.node.XLinkIn)
 camControlIn.setStreamName("camControl")
 camControlIn.out.link(camRgb.inputControl)
 

examples/ColorCamera/rgb_scene.py

@@ -7,8 +7,8 @@
 pipeline = dai.Pipeline()
 
 # Define source and output
-camRgb = pipeline.createColorCamera()
-xoutVideo = pipeline.createXLinkOut()
+camRgb = pipeline.create(dai.node.ColorCamera)
+xoutVideo = pipeline.create(dai.node.XLinkOut)
 
 xoutVideo.setStreamName("video")
 

examples/ColorCamera/rgb_video.py

@@ -8,18 +8,18 @@
 pipeline = dai.Pipeline()
 
 # Define sources and outputs
-camRgb = pipeline.createColorCamera()
-monoLeft = pipeline.createMonoCamera()
-monoRight = pipeline.createMonoCamera()
-
-edgeDetectorLeft = pipeline.createEdgeDetector()
-edgeDetectorRight = pipeline.createEdgeDetector()
-edgeDetectorRgb = pipeline.createEdgeDetector()
-
-xoutEdgeLeft = pipeline.createXLinkOut()
-xoutEdgeRight = pipeline.createXLinkOut()
-xoutEdgeRgb = pipeline.createXLinkOut()
-xinEdgeCfg = pipeline.createXLinkIn()
+camRgb = pipeline.create(dai.node.ColorCamera)
+monoLeft = pipeline.create(dai.node.MonoCamera)
+monoRight = pipeline.create(dai.node.MonoCamera)
+
+edgeDetectorLeft = pipeline.create(dai.node.EdgeDetector)
+edgeDetectorRight = pipeline.create(dai.node.EdgeDetector)
+edgeDetectorRgb = pipeline.create(dai.node.EdgeDetector)
+
+xoutEdgeLeft = pipeline.create(dai.node.XLinkOut)
+xoutEdgeRight = pipeline.create(dai.node.XLinkOut)
+xoutEdgeRgb = pipeline.create(dai.node.XLinkOut)
+xinEdgeCfg = pipeline.create(dai.node.XLinkIn)
 
 edgeLeftStr = "edge left"
 edgeRightStr = "edge right"

examples/EdgeDetector/edge_detector.py

@@ -9,8 +9,8 @@
 pipeline = dai.Pipeline()
 
 # Define sources and outputs
-imu = pipeline.createIMU()
-xlinkOut = pipeline.createXLinkOut()
+imu = pipeline.create(dai.node.IMU)
+xlinkOut = pipeline.create(dai.node.XLinkOut)
 
 xlinkOut.setStreamName("imu")
 

examples/IMU/imu_gyroscope_accelerometer.py

@@ -9,8 +9,8 @@
 pipeline = dai.Pipeline()
 
 # Define sources and outputs
-imu = pipeline.createIMU()
-xlinkOut = pipeline.createXLinkOut()
+imu = pipeline.create(dai.node.IMU)
+xlinkOut = pipeline.create(dai.node.XLinkOut)
 
 xlinkOut.setStreamName("imu")