WATonomous / wato_monorepo

Dockerized ROS2 stack for the WATonomous Autonomous Driving Software Pipeline
Apache License 2.0
84 stars 1 forks source link

Extend Camera Object Detection for multiple cameras within the same node -> Batched inference, tensorRT #151

Open Kishore-Yogaraj opened 1 month ago

Kishore-Yogaraj commented 1 month ago

Need to use TensorRT, something like https://github.com/noahmr/yolov5-tensorrt for yolov8 ➝ https://github.com/triple-Mu/YOLOv8-TensorRT/blob/main/infer-det.py

will be in C++

also look at https://github.com/NVIDIA-AI-IOT/YOLOv5-with-Isaac-ROS

[justin] read 2.1. Batching section here https://docs.nvidia.com/deeplearning/tensorrt/archives/tensorrt-700/pdf/TensorRT-Best-Practices.pdf Approach: There are three subtasks for this:

Arfan12630 commented 1 week ago

Buffer for getting messages from all 6 camera angles https://wiki.ros.org/message_filters#registerCallback.28.29