一种权重自适应与任务解耦的旋转目标检测器
摘要:
本发明提出了一种权重自适应与任务解耦的旋转目标检测器。首先,利用标签信息统计每个类别的数量并计算各自的概率,再对各个类别的概率进行Huffman编码,得到编码权重,在计算预测值和one‑hot标签的BCE损失时,将每个类别的计算结果乘上各自类别的编码权重。其次,将yolov5‑obb检测器的检测头进行解耦操作,采用两个不同的1×1卷积对分类和回归任务进行各自预测,同时,还引入了EIOU损失来代替CIOU提高定位准确性,最终提高遥感场景下目标检测精度。本发明主要应用于遥感场景下的任务精化及长尾目标检测,在遥感目标检测,国防科技及军事等领域具有开阔的应用前景。
0/0