Isaac ROS Image Segmentation

https://media.githubusercontent.com/media/NVIDIA-ISAAC-ROS/.github/main/resources/isaac_ros_docs/repositories_and_packages/isaac_ros_image_segmentation/isaac_ros_image_segmentation_example.png/ https://media.githubusercontent.com/media/NVIDIA-ISAAC-ROS/.github/main/resources/isaac_ros_docs/repositories_and_packages/isaac_ros_image_segmentation/isaac_ros_image_segmentation_example_seg.png/

Overview

Isaac ROS Image Segmentation contains ROS packages for semantic image segmentation.

These packages provide methods for classification of an input image at the pixel level by running GPU-accelerated inference on a DNN model. Each pixel of the input image is predicted to belong to a set of defined classes. The output prediction can be used by perception functions to understand where each class is spatially in a 2D image or fuse with a corresponding depth location in a 3D scene.

https://media.githubusercontent.com/media/NVIDIA-ISAAC-ROS/.github/main/resources/isaac_ros_docs/repositories_and_packages/isaac_ros_image_segmentation/isaac_ros_image_segmentation_nodegraph.png/

Package

Model Architecture

Description

Isaac ROS U-NET

U-NET

Convolutional network popular for biomedical imaging segmentation models

Isaac ROS Segformer

Segformer

Transformer-based network that works well for objects of varying scale

Isaac ROS Segment Anything

Segment Anything

Segments any object in an image when given a prompt as to which one

Input images may need to be cropped and resized to maintain the aspect ratio and match the input resolution expected by the DNN model; image resolution may be reduced to improve DNN inference performance, which typically scales directly with the number of pixels in the image.

https://media.githubusercontent.com/media/NVIDIA-ISAAC-ROS/.github/main/resources/isaac_ros_docs/repositories_and_packages/isaac_ros_image_segmentation/isaac_ros_image_segmentation_example_bboxseg.png/

Image segmentation provides more information and uses more compute than object detection to produce classifications per pixel, whereas object detection classifies a simpler bounding box rectangle in image coordinates. Object detection is used to know if, and where spatially in a 2D image, the object exists. On the other hand, image segmentation is used to know which pixels belong to the class. One application is using the segmentation result, and fusing it with the corresponding depth information in order to know an object location in a 3D scene.

Quickstarts

Isaac ROS NITROS Acceleration

This package is powered by NVIDIA Isaac Transport for ROS (NITROS), which leverages type adaptation and negotiation to optimize message formats and dramatically accelerate communication between participating nodes.

DNN models

Each image segmentation package relies on a pre-trained DNN model. You can train your own models or download pre-trained models from one of the many model zoo’s available online or from NGC which provides pre-trained models for use in your robotics application.

NGC pre-trained models can be fine-tuned for your application using TAO used in the examples provided. See here for more information on how to use various models from NVIDIA hosted on NGC.

U-NET

A U-NET model is required to use isaac_ros_unet.

PeopleSemSegNet provides a pre-trained model for best-in-class, real-time people segmentation.

This package has been validated against the following models:

Model Name

Use Case

PeopleSemSegNet AMR

Segment people from point of view of a mobile robot

PeopleSemSegNet ShuffleSeg

Semantically segment people inference at a high speed

PeopleSemSegNet

Semantically segment people

Segformer

A Segformer model is required to use isaac_ros_segformer.

PeopleSemSegFormer provides a pre-trained model for more effective real-time people segmentation.

This package has been validated against the following models:

Model Name

Use Case

PeopleSemSegFormer

Semantically segment people

Segment Anything

The Segment Anything (SAM) model is required to use isaac_ros_segment_anything. This model requires an additional prompt value to indicate what object in the image should be segmented.

This package has been validated against the following versions of SAM:

Model Name

Description

SAM

Full accuracy model that can consume over 8GB of GPU memory and require more computational power

Mobile SAM

Slimmer variant of SAM optimized for a smaller memory and compute footprint.

Packages

Supported Platforms

This package is designed and tested to be compatible with ROS 2 Humble running on Jetson or an x86_64 system with an NVIDIA GPU.

Note

Versions of ROS 2 other than Humble are not supported. This package depends on specific ROS 2 implementation features that were introduced beginning with the Humble release. ROS 2 versions after Humble have not yet been tested.

Platform

Hardware

Software

Notes

Jetson

Jetson Orin

JetPack 6.0

For best performance, ensure that power settings are configured appropriately.


Jetson Orin Nano 4GB may not have enough memory to run many of the Isaac ROS packages and is not recommended.

x86_64

Ampere or higher NVIDIA GPU Architecture with 8 GB RAM or higher

Ubuntu 22.04+

CUDA 12.2+

Docker

To simplify development, we strongly recommend leveraging the Isaac ROS Dev Docker images by following these steps. This will streamline your development environment setup with the correct versions of dependencies on both Jetson and x86_64 platforms.

Note

All Isaac ROS Quickstarts, tutorials, and examples have been designed with the Isaac ROS Docker images as a prerequisite.

Customize your Dev Environment

To customize your development environment, reference this guide.

Updates

Date

Changes

2024-09-26

Update for ZED compatibility

2024-05-30

Add SegFormer and Segment Anything packages

2023-10-18

Updated for Isaac ROS 2.0.0.

2023-05-25

Performance improvements

2023-04-05

Source available GXF extensions

2022-10-19

Updated OSS licensing

2022-08-31

Update to be compatible with JetPack 5.0.2

2022-06-30

Removed frame_id, queue_size and tensor_output_order parameter. Added network_output_type parameter (support for sigmoid and argmax output layers). Switched implementation to use NITROS. Removed support for odd sized images. Switched tutorial to use PeopleSemSegNet ShuffleSeg and moved unnecessary details to other READMEs.

2021-11-15

Isaac Sim HIL documentation update

2021-10-20

Initial release