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: