From f97e9fa13b48bf3e2890b6cebbdecaf37926af82 Mon Sep 17 00:00:00 2001 From: ricky0614 <162050908+ricky0614@users.noreply.github.com> Date: Mon, 13 Jan 2025 14:54:05 +0900 Subject: [PATCH] Update fast_rcnn.py --- detectron2/modeling/roi_heads/fast_rcnn.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/detectron2/modeling/roi_heads/fast_rcnn.py b/detectron2/modeling/roi_heads/fast_rcnn.py index 0d4d51c744..f1f15d1779 100644 --- a/detectron2/modeling/roi_heads/fast_rcnn.py +++ b/detectron2/modeling/roi_heads/fast_rcnn.py @@ -351,6 +351,9 @@ def losses(self, predictions, proposals): # BCE Loss gt_one_hot = F.one_hot(gt_classes, num_classes=pred_class_logits.size(1)).float() loss_cls = F.binary_cross_entropy_with_logits(pred_class_logits, gt_one_hot, reduction="mean") + elif loss_type == 'dummy': + # dummy loss + loss_cls =　torch.tensor(1.0, requires_grad=True, device=predictions[0].device) elif self.use_sigmoid_ce: loss_cls = self.sigmoid_cross_entropy_loss(scores, gt_classes) else: