yjh0410 il y a 9 mois
Parent
commit
e42ebd29a9
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  1. 1 1
      yolo/models/yolov10/yolov10_head.py

+ 1 - 1
yolo/models/yolov10/yolov10_head.py

@@ -33,7 +33,7 @@ class DetHead(nn.Module):
                 cls_feats.append(ConvModule(in_dim, self.cls_head_dim, kernel_size=1))
             else:
                 cls_feats.append(ConvModule(self.cls_head_dim, self.cls_head_dim, kernel_size=3, stride=1, groups=self.cls_head_dim))
-                cls_feats.append(ConvModule(self.cls_head_dim, self.cls_head_dim, kernel_size=3, stride=1))
+                cls_feats.append(ConvModule(self.cls_head_dim, self.cls_head_dim, kernel_size=1))
         
         ## bbox regression head
         reg_feats = []