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

发布时间:2024年01月08日

在最新的在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
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。