Fix lad repeatedly output warning message (#6584)

open-mmlab · Nov 30, 2021 · c7ef5a4 · c7ef5a4
1 parent a8a1ea3
commit c7ef5a4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmdet/models/dense_heads/lad_head.py b/mmdet/models/dense_heads/lad_head.py
@@ -55,7 +55,7 @@ def get_label_assignment(self,
         """
 
         featmap_sizes = [featmap.size()[-2:] for featmap in cls_scores]
-        assert len(featmap_sizes) == self.anchor_generator.num_levels
+        assert len(featmap_sizes) == self.prior_generator.num_levels
 
         device = cls_scores[0].device
         anchor_list, valid_flag_list = self.get_anchors(