Browse Source

check v10

yjh0410 9 tháng trước cách đây
mục cha
commit
73b1d536a2
1 tập tin đã thay đổi với 1 bổ sung1 xóa
  1. 1 1
      yolo/models/yolov10/yolov10_head.py

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

@@ -118,7 +118,7 @@ class Yolov10DetHead(nn.Module):
                    "pred_reg":       reg_preds,        # List(Tensor) [B, M, 4*(reg_max)]
                    "pred_box":       box_preds,        # List(Tensor) [B, M, 4]
                    "anchors":        anchors,          # List(Tensor) [M, 2]
-                   "stride_tensors": strides,          # List(Tensor) [M, 1]
+                   "stride_tensor": strides,          # List(Tensor) [M, 1]
                    "strides":        self.out_stride,  # List(Int) = [8, 16, 32]
                    }