@@ -167,7 +167,7 @@ class YOLOv2(nn.Module):
cls_feat, reg_feat = self.head(feat)
# 预测层
- obj_pred = self.obj_pred(cls_feat)
+ obj_pred = self.obj_pred(reg_feat)
cls_pred = self.cls_pred(cls_feat)
reg_pred = self.reg_pred(reg_feat)
fmp_size = obj_pred.shape[-2:]
yjh0410