未验证 提交 88e24bf2 编写于 作者: H Hakjin Lee 提交者: GitHub

[Fix] RT-DETR SSOD iou_score (#8409)

上级 02f58862
...@@ -325,15 +325,15 @@ class DETRLoss(nn.Layer): ...@@ -325,15 +325,15 @@ class DETRLoss(nn.Layer):
match_indices = dn_match_indices match_indices = dn_match_indices
if self.use_vfl: if self.use_vfl:
if sum(len(a) for a in gt_bbox) > 0: if gt_score is not None: #ssod
_, target_score = self._get_src_target_assign(
logits[-1].detach(), gt_score, match_indices)
elif sum(len(a) for a in gt_bbox) > 0:
src_bbox, target_bbox = self._get_src_target_assign( src_bbox, target_bbox = self._get_src_target_assign(
boxes.detach(), gt_bbox, match_indices) boxes.detach(), gt_bbox, match_indices)
iou_score = bbox_iou( iou_score = bbox_iou(
bbox_cxcywh_to_xyxy(src_bbox).split(4, -1), bbox_cxcywh_to_xyxy(src_bbox).split(4, -1),
bbox_cxcywh_to_xyxy(target_bbox).split(4, -1)) bbox_cxcywh_to_xyxy(target_bbox).split(4, -1))
if gt_score is not None: #ssod
_, target_score = self._get_src_target_assign(
logits[-1].detach(), gt_score, match_indices)
else: else:
iou_score = None iou_score = None
else: else:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册