YOLOv4 with TensorRT in ROS

Object detection for autonomous robots using state-of-the-art YOLOv4. To optimise models for deployment on Jetson devices, models were serialised into TensorRT engine files for inference. As ROS is one of the most popular middle-ware used for robots, this project performs inference on camera/video input and publishes detection in ROS-supported message formats. This allows anyone to easily modify and use this package in their own projects.

This project supports YOLOv3, tiny-YOLOv3, YOLOv4, tiny-YOLOv4. There are plans to include scaled YOLOv4 in the future.

Hope you like the project and feel free to contribute to it!

results

https://github.com/indra4837/yolov4_trt_ros

2 Likes

Hello,
I wish I could test your program but before I can not find how to do: “CATKIN_IGNORE”.
Can you explain to me how and where to put "CATKIN_IGNORE
Note: This package uses nodes similar to the ros_deep_learning package. Please place a CATKIN_IGNORE in this package to avoid a similar node name error catkin_make

Regards
Mr Owczarek

Hi,

You can simply place an empty file named ‘CATKIN_IGNORE’ inside ros_deep_learning. It is just an empty file which tells catkin not to build the ros_deep_learning package. See attached image for an example.

Do install the dependencies to ensure yolov4_trt_ros package builds successfully. Hopes this solve your issue!

2 Likes

Thanks for your help

Hello,
I have the command “cd ~ / catkin_ws && catkin_make” and below is the error message it cannot find “vision_msgsConfig.cmake” and “vision_msgs-config.cmake”. What to do or reinstall.
I will give you a screenshot

Regards
Mr Owczarekcmake probleme.txt (2.6 KB)

Hi,

You need to also install vision_opencv. Apologies that wasnt mentioned in the README

2 Likes