Skip to content

Commit 72da4cc

Browse files
hhaAndroidZwwWayne
authored andcommitted
Fix lad repeatedly output warning message (open-mmlab#6584)
1 parent 29c1cc5 commit 72da4cc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mmdet/models/dense_heads/lad_head.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ def get_label_assignment(self,
5555
"""
5656

5757
featmap_sizes = [featmap.size()[-2:] for featmap in cls_scores]
58-
assert len(featmap_sizes) == self.anchor_generator.num_levels
58+
assert len(featmap_sizes) == self.prior_generator.num_levels
5959

6060
device = cls_scores[0].device
6161
anchor_list, valid_flag_list = self.get_anchors(

0 commit comments

Comments
 (0)