hukaixuan19970627 / yolov5_obb

yolov5 + csl_label.(Oriented Object Detection)(Rotation Detection)(Rotated BBox)基于yolov5的旋转目标检测
GNU General Public License v3.0
1.81k stars 425 forks source link

OpenCV DNN C++ inference #535

Open blackHorz opened 1 year ago

blackHorz commented 1 year ago

The prediction of an Image using OpenCV DNN with python detect.py works very well. SInce I have little to no experience with C++ and I intent to use it in C++. I would appreciate if anyone can provide a sample code to do so.

Apart from that I did some little digging and found out, I can implement it however I am not sure about this part

_, i = obb_nms(rboxes, scores, iou_thres) this leads to the call of c++ code inds = nms_rotated_ext.nms_rotated(dets_th, scores, iou_thr)

So the question is can one directly use it as it is?

Any help would be much appreciated.

UNeedCryDear commented 1 year ago

cv::dnn::NMSBoxes() supports RotatedRect. image

blackHorz commented 1 year ago

Yes, I have implemented it

David-19940718 commented 1 year ago

Hi, @blackHorz, can you provide me with an implementation code? I really need an implementation code for 'yolov5_obb.' I've been facing some difficulties, and I hope you can help me resolve this issue. My email address is 'wangwei10874@gmail.com'