Merge pull request #256 from GoldenStain/develop

Fix misalignment between SOT and dynamic mode
PaddlePaddle · Feb 19, 2025 · 08b36cf · 08b36cf
2 parents e37222a + e86a2b7
commit 08b36cf
Showing 1 changed file with 6 additions and 9 deletions.
diff --git a/ppdet/modeling/losses/yolov5_loss.py b/ppdet/modeling/losses/yolov5_loss.py
@@ -213,15 +213,12 @@ def yolov5_loss(self, pi, t_cls, t_box, t_indices, t_anchor, balance):
                 # loss_cls = self.BCEcls(ps[:, 5:], t)
 
                 t = paddle.full_like(ps[:, 5:], self.cls_neg_label)
-                if not self.to_static:
-                    t = paddle.put_along_axis(
-                        t,
-                        t_cls.unsqueeze(-1),
-                        values=self.cls_pos_label,
-                        axis=1)
-                else:
-                    for i in range(n):
-                        t[i, t_cls[i]] = self.cls_pos_label
+
+                t = paddle.put_along_axis(
+                    t,
+                    t_cls.unsqueeze(-1),
+                    values=self.cls_pos_label,
+                    axis=1)
 
                 loss_cls = self.BCEcls(ps[:, 5:], t)