/deepdriver

Primary LanguagePythonApache License 2.0Apache-2.0

AWS DeepRacer DeepDriver Project

Overview

The DeepDriver project is a sample application built on top of the existing AWS DeepRacer application uses object detection machine learning model through which the AWS DeepRacer device can identify and interpret traffic signs. For detailed information on DeepDriver project, see DeepDriver project Getting Started section.

License

The source code is released under Apache 2.0.

Demo

deepdriver-demo

Installation

Prerequisites

The AWS DeepRacer device comes with all the pre-requisite packages and libraries installed to run the DeepDriver project. More details about pre installed set of packages and libraries on the DeepRacer, and installing required build systems can be found in the Getting Started section of the AWS DeepRacer Opensource page. DeepDriver project requires the AWS DeepRacer application to be installed on the device as it leverages most of the packages from the core application.

The following are the additional software and hardware requirements to get the DeepDriver project to work on the AWS DeepRacer device.

  1. Download and Optimize the object detection model: Follow the instructions to download and optimize the object detection model and copy it to the required location on the AWS DeepRacer device.

  2. Calibrate the AWS DeepRacer (optional): Follow the instructions to calibrate the mechanics of your AWS DeepRacer Vehicle. This should be done so that the vehicle performance is optimal and it behaves as expected.

  3. Setup Intel Neural Compute Stick 2 (optional): The object_detection_node provides functionality to offload the inference to a Intel Neural Compute Stick 2 connected to the AWS DeepRacer device. This is an optional setting that is provided to enhance the inference performance of the object detection model. More details about running Inference on the Movidius NCS (Neural Compute Stick) with OpenVINO™ toolkit can be found here: https://www.youtube.com/watch?v=XPvMrGobe7I

    Attach the Neural Compute Stick 2 firmly in the back slot of the AWS DeepRacer, and open up a terminal and run the following commands as root user to install the dependencies of the Intel Neural Compute Stick 2 on the AWS DeepRacer device:

    1. Switch to root user:

       sudo su
      
    2. Navigate to the OpenVino installation directory:

       cd /opt/intel/openvino_2021.1.110/install_dependencies
      
    3. Run the dependency installation script for Intel Neural Compute Stick:

       ./install_NCS_udev_rules.sh
      

Downloading and Building

Open up a terminal on the DeepRacer device and run the following commands as root user.

  1. Switch to root user before you source the ROS2 installation:

     sudo su
    
  2. Stop the deepracer-core.service that is currently running on the device:

     systemctl stop deepracer-core
    
  3. Source the ROS2 Foxy setup bash script:

     source /opt/ros/foxy/setup.bash 
    
  4. Set the environment variables required to run Intel OpenVino scripts:

     source /opt/intel/openvino_2021/bin/setupvars.sh
    
  5. Create a workspace directory for the package:

     mkdir -p ~/deepracer_ws
     cd ~/deepracer_ws
    
  6. Clone the entire DeepDriver project on the DeepRacer device.

     git clone https://github.com/jochem725/deepdriver.git
     cd ~/deepracer_ws/deepdriver/deepdriver_ws/
    
  7. Clone the async_web_server_cpp, web_video_server and rplidar_ros dependency packages on the DeepRacer device:

     cd ~/deepracer_ws/deepdriver/deepdriver_ws/ && ./install_dependencies.sh
    
  8. Fetch unreleased dependencies:

     cd ~/deepracer_ws/deepdriver/deepdriver_ws/
     rosws update
    
  9. Resolve the dependencies:

     cd ~/deepracer_ws/deepdriver/deepdriver_ws/ && rosdep install -i --from-path . --rosdistro foxy -y
    
  10. Build the packages in the workspace

     cd ~/deepracer_ws/deepdriver/deepdriver_ws/ && colcon build
    

Usage

Run the node

To launch the DeepDriver sample application as root user on the AWS DeepRacer device open up another terminal on the device and run the following commands as root user:

  1. Switch to root user before you source the ROS2 installation:

     sudo su
    
  2. Source the ROS2 Foxy setup bash script:

     source /opt/ros/foxy/setup.bash 
    
  3. Set the environment variables required to run Intel OpenVino scripts:

     source /opt/intel/openvino_2021/bin/setupvars.sh
    
  4. Source the setup script for the installed packages:

     source ~/deepracer_ws/deepdriver/deepdriver_ws/install/setup.bash
    
  5. Launch the nodes required for traffic project:

     ros2 launch deepdriver_launcher deepdriver_launcher.py
    

Enabling "deepdriver" mode using CLI:

Once the deepdriver_launcher has been kicked-off, open up a adjacent new terminal as root user:

  1. Switch to root user before you source the ROS2 installation:

     sudo su
    
  2. Navigate to the DeepDriver workspace:

     cd ~/deepracer_ws/deepdriver/deepdriver_ws/
    
  3. Source the ROS2 Foxy setup bash script:

     source /opt/ros/foxy/setup.bash
    
  4. Source the setup script for the installed packages:

     source ~/deepracer_ws/deepdriver/deepdriver_ws/install/setup.bash
    
  5. Set the mode of the DeepRacer via ctrl_pkg to "traffic" using the below ros2 service call:

     ros2 service call /ctrl_pkg/vehicle_state deepracer_interfaces_pkg/srv/ActiveStateSrv "{state: 3}"
    
  6. Enable "traffic" mode using the below ros2 service call

     ros2 service call /ctrl_pkg/enable_state deepracer_interfaces_pkg/srv/EnableStateSrv "{is_active: True}"
    

Launch Files

The deepdriver_launcher.py included in this package is the main launcher file that launches all the required nodes for the DeepDriver project. This launcher file also includes the nodes from the AWS DeepRacer core application.

    from launch import LaunchDescription
    from launch_ros.actions import Node


    def generate_launch_description():
        ld = LaunchDescription()
        traffic_sign_node = Node(
            package="traffic_sign_pkg",
            namespace="traffic_sign_pkg",
            executable="traffic_sign_node",
            name="traffic_sign_node",
            parameters=[{"PUBLISH_DISPLAY_OUTPUT": True}],
        )
        object_detection_node = Node(
            package="object_detection_pkg",
            namespace="object_detection_pkg",
            executable="object_detection_node",
            name="object_detection_node",
            parameters=[{"DEVICE": "CPU", "PUBLISH_DISPLAY_OUTPUT": True}],
        )
        deepdriver_navigation_node = Node(
            package="deepdriver_navigation_pkg",
            namespace="deepdriver_navigation_pkg",
            executable="deepdriver_navigation_node",
            name="deepdriver_navigation_node",
        )
        camera_node = Node(
            package="camera_pkg",
            namespace="camera_pkg",
            executable="camera_node",
            name="camera_node",
        )
        ctrl_node = Node(
            package="ctrl_pkg",
            namespace="ctrl_pkg",
            executable="ctrl_node",
            name="ctrl_node",
        )
        deepracer_navigation_node = Node(
            package="deepracer_navigation_pkg",
            namespace="deepracer_navigation_pkg",
            executable="deepracer_navigation_node",
            name="deepracer_navigation_node",
        )
        software_update_node = Node(
            package="deepracer_systems_pkg",
            namespace="deepracer_systems_pkg",
            executable="software_update_node",
            name="software_update_node",
        )
        model_loader_node = Node(
            package="deepracer_systems_pkg",
            namespace="deepracer_systems_pkg",
            executable="model_loader_node",
            name="model_loader_node",
        )
        otg_control_node = Node(
            package="deepracer_systems_pkg",
            namespace="deepracer_systems_pkg",
            executable="otg_control_node",
            name="otg_control_node",
        )
        network_monitor_node = Node(
            package="deepracer_systems_pkg",
            namespace="deepracer_systems_pkg",
            executable="network_monitor_node",
            name="network_monitor_node",
        )
        device_info_node = Node(
            package="device_info_pkg",
            namespace="device_info_pkg",
            executable="device_info_node",
            name="device_info_node",
        )
        battery_node = Node(
            package="i2c_pkg",
            namespace="i2c_pkg",
            executable="battery_node",
            name="battery_node",
        )
        inference_node = Node(
            package="inference_pkg",
            namespace="inference_pkg",
            executable="inference_node",
            name="inference_node",
        )
        model_optimizer_node = Node(
            package="model_optimizer_pkg",
            namespace="model_optimizer_pkg",
            executable="model_optimizer_node",
            name="model_optimizer_node",
        )
        rplidar_node = Node(
            package="rplidar_ros",
            namespace="rplidar_ros",
            executable="rplidarNode",
            name="rplidarNode",
            parameters=[
                {
                    "serial_port": "/dev/ttyUSB0",
                    "serial_baudrate": 115200,
                    "frame_id": "laser",
                    "inverted": False,
                    "angle_compensate": True,
                }
            ],
        )
        sensor_fusion_node = Node(
            package="sensor_fusion_pkg",
            namespace="sensor_fusion_pkg",
            executable="sensor_fusion_node",
            name="sensor_fusion_node",
        )
        servo_node = Node(
            package="servo_pkg",
            namespace="servo_pkg",
            executable="servo_node",
            name="servo_node",
        )
        status_led_node = Node(
            package="status_led_pkg",
            namespace="status_led_pkg",
            executable="status_led_node",
            name="status_led_node",
        )
        usb_monitor_node = Node(
            package="usb_monitor_pkg",
            namespace="usb_monitor_pkg",
            executable="usb_monitor_node",
            name="usb_monitor_node",
        )
        webserver_publisher_node = Node(
            package="webserver_pkg",
            namespace="webserver_pkg",
            executable="webserver_publisher_node",
            name="webserver_publisher_node",
        )
        web_video_server_node = Node(
            package="web_video_server",
            namespace="web_video_server",
            executable="web_video_server",
            name="web_video_server",
        )
        ld.add_action(traffic_sign_node)
        ld.add_action(object_detection_node)
        ld.add_action(deepdriver_navigation_node)
        ld.add_action(camera_node)
        ld.add_action(ctrl_node)
        ld.add_action(deepracer_navigation_node)
        ld.add_action(software_update_node)
        ld.add_action(model_loader_node)
        ld.add_action(otg_control_node)
        ld.add_action(network_monitor_node)
        ld.add_action(device_info_node)
        ld.add_action(battery_node)
        ld.add_action(inference_node)
        ld.add_action(model_optimizer_node)
        ld.add_action(rplidar_node)
        ld.add_action(sensor_fusion_node)
        ld.add_action(servo_node)
        ld.add_action(status_led_node)
        ld.add_action(usb_monitor_node)
        ld.add_action(webserver_publisher_node)
        ld.add_action(web_video_server_node)
        return ld

Configuration File and Parameters

Applies to the object_detection_node

Parameter Name Description
DEVICE (optional) If set as MYRIAD, will use the Intel Compute Stick 2 for inference. Else uses CPU for inference by default, even if removed.
PUBLISH_DISPLAY_OUTPUT Set to True/False if the inference output images need to be published to localhost using web_video_server.

Resources