From c7ef5a408e2b119228290cf356ece13b3d92dc60 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Haian=20Huang=28=E6=B7=B1=E5=BA=A6=E7=9C=B8=29?= <1286304229@qq.com> Date: Wed, 24 Nov 2021 19:03:28 +0800 Subject: [PATCH] Fix lad repeatedly output warning message (#6584) --- mmdet/models/dense_heads/lad_head.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mmdet/models/dense_heads/lad_head.py b/mmdet/models/dense_heads/lad_head.py index d45480b64c9..ce518ad62fe 100644 --- 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(