remove TTA 1 pixel offset (ultralytics#2325)

KMint1819 · Mar 1, 2021 · 497f7e6 · 497f7e6
1 parent d4ef221
commit 497f7e6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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: