search

dlibml / dnn

Deep Neural Network Architectures with dlib
Boost Software License 1.0
17 stars 1 forks source link

Are you familiar for orientation detection(Rotation Detection) in c++, especially in dlib dnn? #3

Closed newforrestgump001 closed 3 years ago

newforrestgump001 commented 3 years ago

Are you familiar for orientation detection(Rotation Detection) in c++? The output is rotated rectangle instead of horizontal bounding box. Thank you a lot for your advice!

arrufat commented 3 years ago

I have never tried myself, but doesn't seem that hard... You would need to modify the YOLO loss to output an extra feature (x, y, w, h, o, r) where r would be the rotation angle, and then train with rotated rectangles.

newforrestgump001 commented 3 years ago

Thank you for your help. BTW, Do you have any plan for YOLOV5?

arrufat commented 3 years ago

No, I don't, what I'd like to do at some point though is to improve the bounding box regression, to be similar to the one in YOLOv4 and others, that is, to use cIOU.

newforrestgump001 commented 3 years ago

OK,I got it.