Update ld_retina.py

HikariTJU · Jan 4, 2022 · 971d0cc · 971d0cc
1 parent ff6386c
commit 971d0cc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmdet/models/dense_heads/ld_retina.py b/mmdet/models/dense_heads/ld_retina.py
@@ -100,7 +100,7 @@ def loss_single(self, cls_score, bbox_pred, anchors, labels, label_weights,
         # assigned_vlr[assigned_vlr > 0] = 1
         vlr_weights = assigned_vlr
         # 剔除正样本
-        neg_weights[labels != self.num_classes] = 0
+        vlr_weights[labels != self.num_classes] = 0
         loss_bbox = self.loss_bbox(
             bbox_pred,
             bbox_targets,