From a9a92aec5ce2de3c56982cc77475b1342c7f3321 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 20 May 2022 12:59:05 +0200
Subject: [PATCH] `cv2.resize` interpolation fix (#7903)

Fix for https://github.com/ultralytics/yolov5/discussions/7901
---
 utils/dataloaders.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/utils/dataloaders.py b/utils/dataloaders.py
index 55279de..23ee2d5 100755
--- a/utils/dataloaders.py
+++ b/utils/dataloaders.py
@@ -663,7 +663,7 @@ class LoadImagesAndLabels(Dataset):
             h0, w0 = im.shape[:2]  # orig hw
             r = self.img_size / max(h0, w0)  # ratio
             if r != 1:  # if sizes are not equal
-                interp = cv2.INTER_LINEAR if self.augment else cv2.INTER_AREA  # random.choice(self.rand_interp_methods)
+                interp = cv2.INTER_LINEAR if (self.augment or r > 1) else cv2.INTER_AREA
                 im = cv2.resize(im, (int(w0 * r), int(h0 * r)), interpolation=interp)
             return im, (h0, w0), im.shape[:2]  # im, hw_original, hw_resized
         else: