modify pred init

yolo/models/gelan/gelan_pred.py

@@ -37,6 +37,9 @@ class SingleLevelPredLayer(nn.Module):
         b = self.reg_pred.bias.view(-1, )
         b.data.fill_(1.0)
         self.reg_pred.bias = torch.nn.Parameter(b.view(-1), requires_grad=True)
+        w = self.reg_pred.weight
+        w.data.fill_(0.)
+        self.reg_pred.weight = torch.nn.Parameter(w, requires_grad=True)
 
     def generate_anchors(self, fmp_size):
         """

yolo/models/yolov1/yolov1_pred.py

@@ -36,6 +36,9 @@ class Yolov1DetPredLayer(nn.Module):
         b = self.reg_pred.bias.view(-1, )
         b.data.fill_(1.0)
         self.reg_pred.bias = torch.nn.Parameter(b.view(-1), requires_grad=True)
+        w = self.reg_pred.weight
+        w.data.fill_(0.)
+        self.reg_pred.weight = torch.nn.Parameter(w, requires_grad=True)
 
     def generate_anchors(self, fmp_size):
         """

yolo/models/yolov2/yolov2_pred.py

@@ -39,6 +39,9 @@ class Yolov2DetPredLayer(nn.Module):
         b = self.reg_pred.bias.view(-1, )
         b.data.fill_(1.0)
         self.reg_pred.bias = torch.nn.Parameter(b.view(-1), requires_grad=True)
+        w = self.reg_pred.weight
+        w.data.fill_(0.)
+        self.reg_pred.weight = torch.nn.Parameter(w, requires_grad=True)
 
     def generate_anchors(self, fmp_size):
         """

yolo/models/yolov3/yolov3_pred.py

@@ -45,6 +45,9 @@ class DetPredLayer(nn.Module):
         b = self.reg_pred.bias.view(-1, )
         b.data.fill_(1.0)
         self.reg_pred.bias = torch.nn.Parameter(b.view(-1), requires_grad=True)
+        w = self.reg_pred.weight
+        w.data.fill_(0.)
+        self.reg_pred.weight = torch.nn.Parameter(w, requires_grad=True)
 
     def generate_anchors(self, fmp_size):
         """

yolo/models/yolov5_af/yolov5_af_pred.py

@@ -41,6 +41,9 @@ class AFDetPredLayer(nn.Module):
         b = self.reg_pred.bias.view(-1, )
         b.data.fill_(1.0)
         self.reg_pred.bias = torch.nn.Parameter(b.view(-1), requires_grad=True)
+        w = self.reg_pred.weight
+        w.data.fill_(0.)
+        self.reg_pred.weight = torch.nn.Parameter(w, requires_grad=True)
 
     def generate_anchors(self, fmp_size):
         """

yolo/models/yolov7_af/yolov7_af_pred.py

@@ -41,6 +41,9 @@ class AFDetPredLayer(nn.Module):
         b = self.reg_pred.bias.view(-1, )
         b.data.fill_(1.0)
         self.reg_pred.bias = torch.nn.Parameter(b.view(-1), requires_grad=True)
+        w = self.reg_pred.weight
+        w.data.fill_(0.)
+        self.reg_pred.weight = torch.nn.Parameter(w, requires_grad=True)
 
     def generate_anchors(self, fmp_size):
         """

yolo/models/yolov8/yolov8_pred.py

@@ -38,6 +38,9 @@ class DetPredLayer(nn.Module):
         b = self.reg_pred.bias.view(-1, )
         b.data.fill_(1.0)
         self.reg_pred.bias = torch.nn.Parameter(b.view(-1), requires_grad=True)
+        w = self.reg_pred.weight
+        w.data.fill_(0.)
+        self.reg_pred.weight = torch.nn.Parameter(w, requires_grad=True)
 
     def generate_anchors(self, fmp_size):
         """