diff --git a/crates/re_viewer/src/depthai/depthai.rs b/crates/re_viewer/src/depthai/depthai.rs index a1827714b99a..232e63f6483d 100644 --- a/crates/re_viewer/src/depthai/depthai.rs +++ b/crates/re_viewer/src/depthai/depthai.rs @@ -611,8 +611,8 @@ fn default_neural_networks() -> Vec> { camera: CameraBoardSocket::CAM_A, }), Some(AiModel { - path: String::from("mobilenet-ssd"), - display_name: String::from("MobileNet SSD"), + path: String::from("yolov6nr3_coco_640x352d"), + display_name: String::from("Yolo V6"), camera: CameraBoardSocket::CAM_A, }), Some(AiModel { diff --git a/rerun_py/depthai_viewer/_backend/device.py b/rerun_py/depthai_viewer/_backend/device.py index 677393690023..019285d86b30 100644 --- a/rerun_py/depthai_viewer/_backend/device.py +++ b/rerun_py/depthai_viewer/_backend/device.py @@ -10,9 +10,7 @@ from depthai_sdk import OakCamera from depthai_sdk.classes.packet_handlers import ComponentOutput from depthai_sdk.components import CameraComponent, NNComponent, StereoComponent -from depthai_sdk.components.camera_helper import ( - getClosestIspScale, -) +from depthai_sdk.components.camera_helper import getClosestIspScale from depthai_sdk.components.tof_component import Component from numpy.typing import NDArray @@ -40,7 +38,11 @@ Message, WarningMessage, ) -from depthai_viewer._backend.packet_handler import DetectionContext, PacketHandler, PacketHandlerContext +from depthai_viewer._backend.packet_handler import ( + DetectionContext, + PacketHandler, + PacketHandlerContext, +) from depthai_viewer._backend.store import Store from depthai_viewer.install_requirements import model_dir @@ -352,7 +354,7 @@ def _create_auto_pipeline_config(self, config: PipelineConfiguration) -> Message if nnet_cam_sock is not None: nnet_cam_sock = nnet_cam_sock.board_socket if nnet_cam_sock is not None: - config.ai_model = ALL_NEURAL_NETWORKS[1] # Mobilenet SSd + config.ai_model = ALL_NEURAL_NETWORKS[1] # Yolo V6 config.ai_model.camera = nnet_cam_sock else: config.ai_model = ALL_NEURAL_NETWORKS[1] diff --git a/rerun_py/depthai_viewer/_backend/device_configuration.py b/rerun_py/depthai_viewer/_backend/device_configuration.py index 1b2e93a691c5..20d032ffe301 100644 --- a/rerun_py/depthai_viewer/_backend/device_configuration.py +++ b/rerun_py/depthai_viewer/_backend/device_configuration.py @@ -89,8 +89,8 @@ def out_queue_name(self) -> str: class AiModelConfiguration(BaseModel): # type: ignore[misc] - display_name: str = "MobileNet SSD" - path: str = "mobilenet-ssd" + display_name: str = "Yolo V6" + path: str = "yolov6nr3_coco_640x352" camera: dai.CameraBoardSocket class Config: @@ -116,8 +116,8 @@ def dict(self, *args, **kwargs): # type: ignore[no-untyped-def] camera=dai.CameraBoardSocket.CAM_A, ), AiModelConfiguration( - path="mobilenet-ssd", - display_name="MobileNet SSD", + path="yolov6nr3_coco_640x352", + display_name="Yolo V6", camera=dai.CameraBoardSocket.CAM_A, ), AiModelConfiguration( diff --git a/rerun_py/depthai_viewer/install_requirements.py b/rerun_py/depthai_viewer/install_requirements.py index c2e0bdbd66e2..a71164f7fdee 100644 --- a/rerun_py/depthai_viewer/install_requirements.py +++ b/rerun_py/depthai_viewer/install_requirements.py @@ -50,7 +50,7 @@ def download_blobs() -> None: os.makedirs(model_dir) models = { "yolov8n_coco_640x352": "depthai", - "mobilenet-ssd": "intel", + "yolov6nr3_coco_640x352": "depthai", "face-detection-retail-0004": "intel", "age-gender-recognition-retail-0013": "intel", "yolov6n_thermal_people_256x192": "depthai",