发明公开
- 专利标题: 一种权重自适应与任务解耦的旋转目标检测器
-
申请号: CN202210209127.9申请日: 2022-03-03
-
公开(公告)号: CN116758363A公开(公告)日: 2023-09-15
- 发明人: 何小海 , 曾王明 , 陈洪刚 , 熊书琪 , 吴晓红 , 王正勇
- 申请人: 四川大学
- 申请人地址: 四川省成都市武侯区一环路南一段24号
- 专利权人: 四川大学
- 当前专利权人: 四川大学
- 当前专利权人地址: 四川省成都市武侯区一环路南一段24号
- 主分类号: G06V10/774
- IPC分类号: G06V10/774 ; G06V10/764 ; G06V10/766 ; G06V10/82 ; G06V20/10 ; G06V20/13 ; G06N3/0464 ; G06N3/047 ; G06N3/048 ; G06N3/084 ; G06N3/096
摘要:
本发明提出了一种权重自适应与任务解耦的旋转目标检测器。首先,利用标签信息统计每个类别的数量并计算各自的概率,再对各个类别的概率进行Huffman编码,得到编码权重,在计算预测值和one‑hot标签的BCE损失时,将每个类别的计算结果乘上各自类别的编码权重。其次,将yolov5‑obb检测器的检测头进行解耦操作,采用两个不同的1×1卷积对分类和回归任务进行各自预测,同时,还引入了EIOU损失来代替CIOU提高定位准确性,最终提高遥感场景下目标检测精度。本发明主要应用于遥感场景下的任务精化及长尾目标检测,在遥感目标检测,国防科技及军事等领域具有开阔的应用前景。