Support keeping image ratio in the multi-scale training of YOLOX (ope…

…n-mmlab#6732) * Support keeping ratio in the multi-scale training of YOLOX * fix typos * fix typos
ZwwWayne · Jul 18, 2022 · 21a80ef · 21a80ef
1 parent 274452b
commit 21a80ef
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/mmdet/models/detectors/yolox.py b/mmdet/models/detectors/yolox.py
@@ -119,7 +119,10 @@ def _random_resize(self):
 
         if self.rank == 0:
             size = random.randint(*self._random_size_range)
-            size = (self._size_multiplier * size, self._size_multiplier * size)
+            aspect_ratio = float(
+                self._default_input_size[1]) / self._default_input_size[0]
+            size = (self._size_multiplier * size,
+                    self._size_multiplier * int(aspect_ratio * size))
             tensor[0] = size[0]
             tensor[1] = size[1]