AlexeyAB / darknet

YOLOv4 / Scaled-YOLOv4 / YOLO - Neural Networks for Object Detection (Windows and Linux version of Darknet )
http://pjreddie.com/darknet/
Other
21.68k stars 7.96k forks source link

Further processing on bounding boxes before drawing them #1312

Open teaddicted opened 6 years ago

teaddicted commented 6 years ago

Hey,

I found the functions "draw_label" & "draw_box" inside the "image.c" file. I have trained a binary cnn (in python ofc) from cropped bounding boxes images which I got from your crop code and it's is working absolutely fine . Now my requirement is to do it in darknet (real time). Like before darknet shows the bounding boxes & labels, I need it to crop the bounding box, predict further with cnn and then call the draw_label & draw_box functions with the new predicted label. Any way this can be done?

Thanks a lot.

teaddicted commented 6 years ago

nevermind i got the solution