search

dlibml / darknet

Darknet related stuff: ports of YOLO models to dlib
Boost Software License 1.0
5 stars 1 forks source link

Is the yolo4x incomplete? #8

Closed newforrestgump001 closed 2 years ago

newforrestgump001 commented 2 years ago

I merged yolov4 code snippet to dnn_yolo_train_ex.cpp in dlib, and it works fine. But there is no classes parameter in yolov4x, is it incomplete or maybe I misunderstand it. Thank you!

arrufat commented 2 years ago

I am not sure I understand your question… The model is complete, but it's defined a bit differently than the one in the example… it's already setup to have 80 classes.

arrufat commented 2 years ago

After having a second look at the code, you can just add a template parameter to the yolov4x definition to be the number of classes, and replace all 255 filters with (num_classes + 5) * 3 to have a similar interface as the other ones.

newforrestgump001 commented 2 years ago

Thank you! I am not so familiar to dlib dnn, although very interested with it.