/jetson-ros-vision

Machine Vision for NVIDIA Jetson. GPU accelerated video capture and neural networks for real-time object detection.

Primary LanguageC

ssd_mobilenet_1 ssd_mobilenet_2

Embedded Object-Detection at 40 FPS using MobileNetV2 SSD Neural Network and ROS on Jetson Nano.

Real-time object detection"on-the-edge" at 40 FPS from 720p video streams.

Jetson Nano is a very low power device but is equipped with an NVIDIA GPU. TensorRT can be used to optimize a neural network for GPU achieving enough performance improvement and power efficiency to run inference on the Nano in real-time. TensorRT is built on CUDA, NVIDIA’s parallel programming model, providing optimized inference for artificial intelligence, autonomous machines, high-performance computing, and graphics using graph optimizations, kernel fusion, and quantization.

NVIDIA Jetson Nano

Specifications:

  • GPU: 128-core Maxwell
  • CPU: Quad-core ARM A57 @ 1.43 GHz
  • Memory: 4 GB 64-bit LPDDR4 25.6 GB/s

The NVIDIA Jetson Nano is a low-popwered embedded systems aimed at accelerating machine learning applictions. These can include robotics, automonous systems and smart devices. The Nano is the most constrained of the Jetson series of devices and offers the weakest performance but by careful design it can acheive realtime inference.

jetson_nano

Quickstart

The example is built using Robotic Operating System (ROS2) to provide a modular structure, interprocess communication and a distributed parameter system. Video frames are captured at 1280x720 from the CSI camera with a GStreamer pipeline and are color converted from raw NVMM video data from YuV to RGB using CUDA before being passed upstream. In a prior step a pre-trained PyTorch model is converted to UFF format so that it can be imported into TensorRT. After that the inference takes place entirely on the GPU and uses GPU RAM. The output of inference (bounding boxes for the detected objects and associated confidence level) is sent to a OpenGL display accelerated with CUDA interop. At each stage buffers are used to improve throughput.

Building

From the project root presuming a Jetpack install on the Nano with ROS eloquent

source /opt/ros/eloquent/setup.bash 
source ./install/setup.bash

Run single-node at ~40 FPS

ros2 run jetson-ros-vision single_node --ros-args --params-file ./config/params.yaml

Run multi-node at ~20 FPS

ros2 run jetson-ros-vision multi_node --ros-args --params-file ./config/params.yaml

Note: The first execution will parse the UFF file and create the TensorRT engine which will take some time. Subsequently the engine will be loaded from cache and startup will be quicker but still not fast!

Object Detection

SSD

Object detection is a computer vision technique that allows simultaneous identification and localization of objects in images. When applied to video streams this identification and localization can be used to count objects in a scene and to determine and track their precise locations. This is a task our visual cortex achieves this effortlessly it is computationaly intensive and any CPU will struggle to achieve a 30 FPS real-time inference rate. Fortunately the parallel structure of GPU can help us attain real-time performance even on embedded systems.

Single-Shot-Detectors (SSDs) are a type of neural network that use a set of predetermined regions to detect objects. A grid of anchor points is laid over the input image, and at each anchor point boxes of various dimensions are defined. For each box at each anchor point, the model outputs a prediction of whether or not an object exists within the region. Because there are multiple boxes at each anchor point and anchor points may be close together, SSDs produce detections that overlap. Post-processing (non-maximum suppression) is applied in order to prune away most predictions and pick the best one. This is a one-pass operation which contrast from the two-pass operation of R-CNN. The accuracy of two-pass models is generally better butone-pass models win in terms of speed and are thus attractive in embedded systems.

The SSD has two components:

  • The Backbone Model that is a pre-trained image classification network (e.g. MobileNetV2) from which the final fully connected classification layer has been removed and that acts as a feature extracto.
  • The SSD Head that is just a series of convolutional layers added to the backbone and the outputs are interpreted as the bounding boxes and classes of objects.

TensorRT Networks

TensorRT is NVIDIA’s highly optimized neural network inference framework which works on NVIDIA GPUs. TensorRT speeds up the network by using FP16 and INT8 precision instead of the default FP3 and uses the tensor cores of the GPU instead of the regular CUDA cores.

TensorRT workflow:

  • Create a network description graph consisting of TensorRT layers: here we import an existing network in UFF using the parser.
  • Build a TensorRT runtime engine which optimizes the network for the specific GPU and serialized to disk for later inference.
  • Create a TensorRT execution context specifying and dimensions left “dynamic” in the engine.
  • Use the execution context to run the network.

Future Work

This example provides a basis onto which further optimization and embedded vision tasks can be built. Examples are

  • Support models in ONNX format
  • Improve intra-process communication (compression/down-scaling)
  • Custom trained models
  • Impliment SIFT/SURF/ORB