diff --git a/ultralytics/nn/modules/head.py b/ultralytics/nn/modules/head.py index 5e6af102e..4d6816e34 100644 --- a/ultralytics/nn/modules/head.py +++ b/ultralytics/nn/modules/head.py @@ -42,11 +42,6 @@ def __init__(self, nc=80, ch=()): self.cv3 = nn.ModuleList(nn.Sequential(Conv(x, c3, 3), Conv(c3, c3, 3), nn.Conv2d(c3, self.nc, 1)) for x in ch) self.dfl = DFL(self.reg_max) if self.reg_max > 1 else nn.Identity() - def generate_static_anchors(self, x): - shape = x[0].shape - self.anchors, self.strides = (x.transpose(0, 1) for x in make_anchors(x, self.stride, 0.5)) - self.shape = shape - def inference(self, x): # Inference path shape = x[0].shape # BCHW