在最新的在ultralytics中,官方团队已经提供了obb检测功能,并发布了相应的预训练模型,其具体使用可以参考。https://blog.csdn.net/a486259/article/details/134557389
1、网络结构
1.1 OBB(Detect)
在ultralytics\nn\modules\head.py中,实现了OBB head,用于在网络head中新增预测值用于进行角度预测,可见其是通过cv4 进行角度特征提取,最终预测一个角度值(ne ),非csl、kdl的方式进行预测,其预测的角度范围为[-pi/4, 3pi/4]
class OBB(Detect):
"""YOLOv8 OBB detection head for de