Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
jameslahm committed May 23, 2024
1 parent ccb5ac8 commit 1d81b4a
Showing 1 changed file with 0 additions and 5 deletions.
5 changes: 0 additions & 5 deletions ultralytics/nn/modules/head.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 1d81b4a

Please sign in to comment.