remove TTA 1 pixel offset (#2325)
Browse files- 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] -
|
113 |
elif fi == 3:
|
114 |
-
yi[..., 0] = img_size[1] -
|
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:
|