diff --git a/models.py b/models.py index adbee6ef..45899158 100644 --- a/models.py +++ b/models.py @@ -419,10 +419,10 @@ def __init__(self, yolov4conv137weight=None, n_classes=80, inference=False): self.down4 = DownSample4() self.down5 = DownSample5() # neck - self.neck = Neck(inference) + self.neek = Neck(inference) # yolov4conv137 if yolov4conv137weight: - _model = nn.Sequential(self.down1, self.down2, self.down3, self.down4, self.down5, self.neck) + _model = nn.Sequential(self.down1, self.down2, self.down3, self.down4, self.down5, self.neek) pretrained_dict = torch.load(yolov4conv137weight) model_dict = _model.state_dict() @@ -443,7 +443,7 @@ def forward(self, input): d4 = self.down4(d3) d5 = self.down5(d4) - x20, x13, x6 = self.neck(d5, d4, d3) + x20, x13, x6 = self.neek(d5, d4, d3) output = self.head(x20, x13, x6) return output