yjh0410 hace 9 meses
padre
commit
e42ebd29a9
Se han modificado 1 ficheros con 1 adiciones y 1 borrados
  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 = []