Bladeren bron

check v10

yjh0410 9 maanden geleden
bovenliggende
commit
e42ebd29a9
1 gewijzigde bestanden met toevoegingen van 1 en 1 verwijderingen
  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 = []