我在图像中有一个目标对象和许多其他对象。目标对象预先定义好一个已知的形状，如图所示。
我的任务是检测图像中存在的所有目标对象，并找到检测到的对象相对于目标对象的角度。出于对象检测的目的，我使用了 YOLO-V5-OBB 模型，它给了我检测信心和旋转的边界框坐标。见下面的结果
我想知道 yolo-obb 模型如何预测旋转角度，以便围绕检测到的对象制作旋转边界框？
标签： data-science artificial-intelligence angle yolov5 image-comparison
寻找对象：
在使用重型机器学习模型之前，请尝试使用经典的计算机视觉算法。
寻找对象：
如果上面的对象是您要搜索的唯一对象：
使用cv2.HoughCircles()。
https://docs.opencv.org/3.4/d4/d70/tutorial_hough_circle.html
如果您希望能够搜索任意对象： 尝试使用模板匹配。
https://pyimagesearch.com/2021/03/22/opencv-template-matching-cv2-matchtemplate/
检测到物体后：
应用霍夫变换提取顶线并使用线拟合算法检测角度。
OpenCV line-fitting（可能已弃用）
【讨论】：
我想你可以尝试使用 YOLOv5-OBB 来训练并得到边界框的坐标，然后很容易找到方向的角度。查看detect.py。
# Write results
for *poly, conf, cls in reversed(det):
if save_txt: # Write to file
# xywh =
(xyxy2xywh(torch.tensor(xyxy).view(1, 4)) / gn).view(-1).tolist()
# normalized xywh
【讨论】：