From ef8c59cf50733627563d49ffa8d0444f9c7e588b Mon Sep 17 00:00:00 2001 From: Francesco Mattioli Date: Wed, 30 Oct 2024 15:38:32 +0100 Subject: [PATCH] fixed head format --- ultralytics/nn/modules/head.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ultralytics/nn/modules/head.py b/ultralytics/nn/modules/head.py index 263be0a17c..dba6cacf3a 100644 --- a/ultralytics/nn/modules/head.py +++ b/ultralytics/nn/modules/head.py @@ -28,7 +28,7 @@ class Detect(nn.Module): shape = None anchors = torch.empty(0) # init strides = torch.empty(0) # init - format = "_" + format = None legacy = False # backward compatibility for v3/v5/v8/v9 models def __init__(self, nc=80, ch=()):