yjh0410 9 сар өмнө
parent
commit
73b1d536a2

+ 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]
                    }