search

LdDl / rust-road-traffic

Vehicle counting/tracking and speed estimation
17 stars 3 forks source link

[QUESTION] How to train Traditional YOLO #27

Closed benjamingb closed 3 weeks ago

benjamingb commented 3 weeks ago

Hi How do you train models for Darknet? I've been training it on Ultralytics and converted it to ONNX, but the model is very heavy and also consumes more CPU. Darknet models seem to be lighter, but I haven't found a way to train a custom model. How do you do it?

LdDl commented 3 weeks ago

Hi I prefer to use this README - https://github.com/AlexeyAB/darknet?tab=readme-ov-file#how-to-train-to-detect-your-custom-objects due this is considered to be "classic"

It is pretty good for both novice and advanced user. But be prepared to face installation steps also. You could use Pytorch version with further ONNX conversion, but I've never used it - https://github.com/Tianxiaomo/pytorch-YOLOv4/blob/master/train.py