From bef389dec7d52a9f0312d63458b82e6be39868b4 Mon Sep 17 00:00:00 2001 From: Francis Lata Date: Sun, 19 Jan 2025 15:59:28 -0800 Subject: [PATCH] realize boxcoder's encoding --- extra/models/retinanet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/extra/models/retinanet.py b/extra/models/retinanet.py index 7eddbc51a6..baa77e64ad 100644 --- a/extra/models/retinanet.py +++ b/extra/models/retinanet.py @@ -176,7 +176,7 @@ class RegressionHead: def _compute_loss(self, x:Tensor, bboxes:Tensor, matches:Tensor, anchors:Tensor) -> Tensor: mask = (fg_idxs := matches >= 0).reshape(matches.shape[0], -1, 1) - tgt = self.box_coder.encode(bboxes, anchors) + tgt = self.box_coder.encode(bboxes, anchors).realize() loss = mask.where(l1_loss(x, tgt), 0).sum(-1).sum(-1) loss = (loss / fg_idxs.sum(-1)).sum() / matches.shape[0] return loss