在ultralytics中实现obb检测，官方团队做了哪些修改？

2024-01-08 06:27:55

在最新的在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

文章来源:https://blog.csdn.net/a486259/article/details/135436105
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。