From 417a2f425cd553c1f2a41374a6cd2710aa91d5f0 Mon Sep 17 00:00:00 2001
From: Yonghye Kwon <developer.0hye@gmail.com>
Date: Thu, 24 Jun 2021 22:57:27 +0900
Subject: [PATCH] Edit comment (#3759)

edit comment
---
 utils/datasets.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/utils/datasets.py b/utils/datasets.py
index abb4a3650bfc..d3714d745b88 100755
--- a/utils/datasets.py
+++ b/utils/datasets.py
@@ -574,7 +574,7 @@ def __getitem__(self, index):
             labels_out[:, 1:] = torch.from_numpy(labels)
 
         # Convert
-        img = img[:, :, ::-1].transpose(2, 0, 1)  # BGR to RGB, to 3x416x416
+        img = img[:, :, ::-1].transpose(2, 0, 1)  # BGR to RGB, to 3 x img_height x img_width
         img = np.ascontiguousarray(img)
 
         return torch.from_numpy(img), labels_out, self.img_files[index], shapes