2.Stereo Depth Algorithm

Function Introduction

The Digua stereo depth estimation algorithm takes stereo image data as input and outputs a disparity map and a depth map corresponding to the left view. The algorithm draws inspiration from the IGEV network and employs a GRU architecture, offering good data generalization and high inference efficiency.

RDK Official Stereo Depth Algorithm Tutorial：Stereo Depth Algorithm | RDK DOC

Stereo algorithm repository：https://github.com/D-Robotics/hobot_stereonet

MIPI camera repository：https://github.com/D-Robotics/hobot_mipi_cam

Supported Platforms

Algorithm Version Information

The following versions of the stereo algorithm are currently available:

RDK Platform

1.Confirm that the camera is correctly connected to the RDK board.

The installation method is shown in the figure. Do not reverse the wiring, otherwise the left and right diagrams will be swapped, and the binocular algorithm will malfunction.

RDK X5 access method

S100(P) access method

With the power off, the camera expansion board switches are set to LPWM and 3.3V respectively.

Camera wiring

2.The RDK has already burned the Ubuntu 22.04 system image.

3.RDK has successfully installed tros.b

4.Confirm that the PC can access RDK via the network.

System and Package Versions

• If the system version does not meet the requirements, please refer to the corresponding documentation section for image flashing.
• If the package version does not meet the requirements, please execute the following commands to update:

sudo apt update
sudo apt upgrade

Usage Instructions

NOTE：Please execute the commands in this document as the root user. Other users may encounter permission issues, leading to unnecessary errors.

Confirm camera connection

• To verify a proper camera connection, connect to RDK via SSH and execute the following command. If addresses such as 0x30, 0x32, or 0x50 are output, the camera connection is successful:

RDK X5 connection

xxxxxxxxxx
# RDK X5
i2cdetect -r -y 4
i2cdetect -r -y 6

S100(P) connection

xxxxxxxxxx
# RDK S100
i2cdetect -r -y 1
i2cdetect -r -y 2

Start the stereo algorithm

• To launch the corresponding version of the stereo algorithm using different launch files, connect to the RDK via SSH and execute the following commands:

RDK X5

The following commands only require selecting one version to run; after running them, continue reading.

​x
# Only supports RDK X5
​
# Configure the tros.b humble environment
source /opt/tros/humble/setup.bash
​
# The launch file for the stereo model contains the startup instructions for the algorithm and the stereo camera nodes.
ros2 launch hobot_stereonet stereonet_model_web_visual_v2.0.launch.py \
mipi_image_width:=640 mipi_image_height:=352 mipi_lpwm_enable:=True mipi_image_framerate:=30.0 mipi_rotation:=90.0 \
need_rectify:=False height_min:=-10.0 height_max:=10.0 pc_max_depth:=5.0

xxxxxxxxxx
# Supports RDK X5 and RDK s100(p)
​
# Configure the tros.b humble environment
source /opt/tros/humble/setup.bash
​
# The launch file for the stereo model contains the startup instructions for the algorithm and the stereo camera nodes.
ros2 launch hobot_stereonet stereonet_model_web_visual_v2.1.launch.py \
mipi_image_width:=640 mipi_image_height:=352 mipi_lpwm_enable:=True mipi_image_framerate:=30.0 mipi_rotation:=90.0 \
need_rectify:=False height_min:=-10.0 height_max:=10.0 pc_max_depth:=5.0 \
uncertainty_th:=0.1

xxxxxxxxxx
# Only supports RDK X5
​
# Configure the tros.b humble environment
source /opt/tros/humble/setup.bash
​
# The launch file for the stereo model contains the startup instructions for the algorithm and the stereo camera nodes.
ros2 launch hobot_stereonet stereonet_model_web_visual_v2.2.launch.py \
mipi_image_width:=640 mipi_image_height:=352 mipi_lpwm_enable:=True mipi_image_framerate:=30.0 mipi_rotation:=90.0 \
need_rectify:=False height_min:=-10.0 height_max:=10.0 pc_max_depth:=5.0

xxxxxxxxxx
# Only supports RDK X5
​
# Configure the tros.b humble environment
source /opt/tros/humble/setup.bash
​
# The launch file for the stereo model contains the startup instructions for the algorithm and the stereo camera nodes.
ros2 launch hobot_stereonet stereonet_model_web_visual_v2.3.launch.py \
mipi_image_width:=640 mipi_image_height:=352 mipi_lpwm_enable:=True mipi_image_framerate:=30.0 mipi_rotation:=90.0 \
need_rectify:=False height_min:=-10.0 height_max:=10.0 pc_max_depth:=5.0

xxxxxxxxxx
# RDK X5 executes the following commands:
​
# Select either int16 or int8 version
# Configure the tros.b humble environment
source /opt/tros/humble/setup.bash
​
# The int16 version is the launch file for starting the stereo model, which contains the startup instructions for the algorithm and stereo camera nodes.
ros2 launch hobot_stereonet stereonet_model_web_visual_v2.4_int16.launch.py \
mipi_image_width:=640 mipi_image_height:=352 mipi_lpwm_enable:=True mipi_image_framerate:=30.0 mipi_rotation:=90.0 \
need_rectify:=False height_min:=-10.0 height_max:=10.0 pc_max_depth:=5.0
​
# This is an int8 version launcher file for the stereo model, containing the startup instructions for the algorithm and stereo camera nodes.
ros2 launch hobot_stereonet stereonet_model_web_visual_v2.4_int8.launch.py \
mipi_image_width:=640 mipi_image_height:=352 mipi_lpwm_enable:=True mipi_image_framerate:=30.0 mipi_rotation:=90.0 \
need_rectify:=False height_min:=-10.0 height_max:=10.0 pc_max_depth:=5.0
​
# The RDK S100 executes the following command:
​
# Configure the tros.b humble environment
source /opt/tros/humble/setup.bash
​
# The launch file for the stereo model contains the startup instructions for the algorithm and the stereo camera nodes.
​
ros2 launch hobot_stereonet stereonet_model_web_visual_v2.4.launch.py \
mipi_image_width:=640 mipi_image_height:=352 mipi_lpwm_enable:=True mipi_image_framerate:=30.0 mipi_rotation:=90.0 \
need_rectify:=False height_min:=-10.0 height_max:=10.0 pc_max_depth:=5.0 \
uncertainty_th:=0.1

Parameter descriptions:

• The following log indicates that the stereo algorithm has started successfully. fx/fy/cx/cy/base_line are the camera intrinsic parameters. If the depth map is normal but the estimated distance is incorrect, there may be an issue with the camera intrinsic parameters:

View the depth map

• View the depth map via the web interface by entering http://ip:8000 in the browser (the RDK IP in the image is 192.168.1.100):

View the point cloud

• View the point cloud via rviz2. RDK can directly install rviz2 for viewing. Note the following configurations in rviz2:

xxxxxxxxxx
# Install rviz2
sudo apt install ros-humble-rviz2
# Start rviz2
source /opt/tros/humble/setup.bash
rviz2

• If the user wants to save the depth estimation results, the following parameters can be added to enable saving. save_image_all enables saving, save_freq controls the saving frequency, save_dir specifies the save directory (automatically created if it does not exist), and save_total controls the total number of saves. The program will save camera intrinsic parameters, left and right images, disparity map, depth map, and visualization map:

xxxxxxxxxx
# Configure tros.b humble environment
source /opt/tros/humble/setup.bash
​
# Here, the V2.0 version of the algorithm is used as an example. Similar parameters can be added to other versions of the algorithm.
ros2 launch hobot_stereonet stereonet_model_web_visual_v2.0.launch.py \
mipi_image_width:=640 mipi_image_height:=352 mipi_lpwm_enable:=True mipi_image_framerate:=30.0 \
need_rectify:=False height_min:=-10.0 height_max:=10.0 pc_max_depth:=5.0 \
save_image_all:=True save_freq:=4 save_dir:=./online_result save_total:=10

Parameter descriptions: