search

atom-robotics-lab / ros-perception-pipeline

This is a ROS 2 based plug-and-play perception pipeline which can be customised to your need
https://atom-robotics-lab.github.io/
8 stars 9 forks source link

Re-write YOLOv5 Detector code using Ultralytics #57

Closed topguns837 closed 9 months ago

topguns837 commented 9 months ago

Problem Description The code for YOLOv5 Detector uses OpenCV's DNN module. To use GPU with this detector, we will need to build OpenCV from source which is cumbersome inside a docker container.

Rewrite the whole code using ultralytics. This will reduce the lines of code and make it future proof