search

leggedrobotics / darknet_ros

YOLO ROS: Real-Time Object Detection for ROS
BSD 3-Clause "New" or "Revised" License
2.14k stars 1.16k forks source link

YOLOv4 support #328

Closed enekorodriguezplaza closed 3 years ago

enekorodriguezplaza commented 3 years ago

Dear all,

I am wondering whether this repository can also be used for yolov4 as well. If this was the case, what files exactly should be changed? Could I just port my .cfg and weights obtained from the darknet repo, and then simply add the weights found?

Kind regards, Eneko

Ar-Ray-code commented 3 years ago

Hi.

I know of a few darknet_ros repositories that support YOLO v4, so I'll introduce them to you. Are you using ROS1 or ROS2?

Note : To support YOLO v4, AlexeyAB/darknet needs to be prepared as a submodule.

ROS1 Melodic ( I haven't done any testing with Ubuntu20.)

https://github.com/Tossy0423/yolov4-for-darknet_ros

ROS2 Foxy ( I haven't done any testing with Ubuntu18.)

https://github.com/Ar-Ray-code/darknet_ros_yolov4

I've already opened a pull request.

I hope it works.

enekorodriguezplaza commented 3 years ago

I'm using ROS1 witrh Ubuntu 18, so I'll try the Tossy repo. Thanks a lot for the tip!

tomlankhorst commented 3 years ago

Hope this resolves your issue, feel free to reopen otherwise.