rhysdg / ros2_torch2trt_examples

ros2 packages for torch2trt examples

ros2_torch2trt_examples

The repository contains ros2 nodes for torch2trt examples. torch2trt is a PyTorch to TensorRT converter which utilizes the TensorRT Python API.

Following examples are currently supported in ROS2:
- MiDAS: Robust Monocular Depth Estimation
- EAST: An Efficient and Accurate Scene Text Detector

Requirements

• ROS2 Eloquent
  

  • Docker Support
    
  • Install Scripts
    

• torch2trt

  • Install torch2trt

Please use ReadMe for each package for built, and run instructions.