Use num_base_priors to init bias instead of 3 (#658)

2023-03-14 09:49:20 +08:00 · 2023-03-14 09:49:20 +08:00 · 6341f0e31b
parent 71fc5670cb
commit 6341f0e31b
1 changed files with 1 additions and 1 deletions
--- a/mmyolo/models/dense_heads/yolov7_head.py
+++ b/mmyolo/models/dense_heads/yolov7_head.py
@ -39,7 +39,7 @@ class YOLOv7HeadModule(YOLOv5HeadModule):
        for mi, s in zip(self.convs_pred, self.featmap_strides):  # from
            mi = mi[1]  # nn.Conv2d

-            b = mi.bias.data.view(3, -1)
+            b = mi.bias.data.view(self.num_base_priors, -1)
            # obj (8 objects per 640 image)
            b.data[:, 4] += math.log(8 / (640 / s)**2)
            b.data[:, 5:] += math.log(0.6 / (self.num_classes - 0.99))