diff --git a/mmdet/models/dense_heads/rr_yolov5_head.py b/mmdet/models/dense_heads/rr_yolov5_head.py
index b27f3bed..44a94464 100644
--- a/mmdet/models/dense_heads/rr_yolov5_head.py
+++ b/mmdet/models/dense_heads/rr_yolov5_head.py
@@ -70,9 +70,9 @@ def _init_layers(self):
 
         self.det = nn.Sequential(*model)
         self.head = nn.Sequential(
-            nn.Conv2d(make_div8_fun(256), 255, 1),
-            nn.Conv2d(make_div8_fun(512), 255, 1),
-            nn.Conv2d(make_div8_fun(1024), 255, 1),
+            nn.Conv2d(make_div8_fun(256), (self.num_classes + 5) * 3, 1),
+            nn.Conv2d(make_div8_fun(512), (self.num_classes + 5) * 3, 1),
+            nn.Conv2d(make_div8_fun(1024), (self.num_classes + 5) * 3, 1),
         )
 
     def forward(self, feats):