fixed head format

pull/17463/head^2
Francesco Mattioli 1 month ago
parent b339394537
commit ef8c59cf50
  1. 2
      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=()):

Loading…
Cancel
Save