glenn-jocher commited on
Commit
fd96810
·
unverified ·
1 Parent(s): c2026a5

remove TTA 1 pixel offset (#2325)

Browse files
Files changed (1) hide show
  1. models/yolo.py +2 -2
models/yolo.py CHANGED
@@ -109,9 +109,9 @@ class Model(nn.Module):
109
  # cv2.imwrite(f'img_{si}.jpg', 255 * xi[0].cpu().numpy().transpose((1, 2, 0))[:, :, ::-1]) # save
110
  yi[..., :4] /= si # de-scale
111
  if fi == 2:
112
- yi[..., 1] = img_size[0] - 1 - yi[..., 1] # de-flip ud
113
  elif fi == 3:
114
- yi[..., 0] = img_size[1] - 1 - yi[..., 0] # de-flip lr
115
  y.append(yi)
116
  return torch.cat(y, 1), None # augmented inference, train
117
  else:
 
109
  # cv2.imwrite(f'img_{si}.jpg', 255 * xi[0].cpu().numpy().transpose((1, 2, 0))[:, :, ::-1]) # save
110
  yi[..., :4] /= si # de-scale
111
  if fi == 2:
112
+ yi[..., 1] = img_size[0] - yi[..., 1] # de-flip ud
113
  elif fi == 3:
114
+ yi[..., 0] = img_size[1] - yi[..., 0] # de-flip lr
115
  y.append(yi)
116
  return torch.cat(y, 1), None # augmented inference, train
117
  else: