From 497f7e6cdc103705c24abb41cbf5faa649f55c0f Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Sun, 28 Feb 2021 21:14:08 -0800
Subject: [PATCH] remove TTA 1 pixel offset (#2325)

---
 models/yolo.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/models/yolo.py b/models/yolo.py
index 85043f2b0205..a9e1da43d913 100644
--- a/models/yolo.py
+++ b/models/yolo.py
@@ -109,9 +109,9 @@ def forward(self, x, augment=False, profile=False):
                 # cv2.imwrite(f'img_{si}.jpg', 255 * xi[0].cpu().numpy().transpose((1, 2, 0))[:, :, ::-1])  # save
                 yi[..., :4] /= si  # de-scale
                 if fi == 2:
-                    yi[..., 1] = img_size[0] - 1 - yi[..., 1]  # de-flip ud
+                    yi[..., 1] = img_size[0] - yi[..., 1]  # de-flip ud
                 elif fi == 3:
-                    yi[..., 0] = img_size[1] - 1 - yi[..., 0]  # de-flip lr
+                    yi[..., 0] = img_size[1] - yi[..., 0]  # de-flip lr
                 y.append(yi)
             return torch.cat(y, 1), None  # augmented inference, train
         else: