🔨 [Add] image ratio min-max constrain

yolo/tools/data_loader.py

@@ -141,7 +141,7 @@ class YoloDataset(Dataset):
         if bboxes:
             return torch.stack(bboxes)
         else:
-            logger.warning("No valid BBox in {}", label_path)
+            logger.warning(f"No valid BBox in {label_path}")
             return torch.zeros((0, 5))
 
     def get_data(self, idx):
@@ -158,8 +158,7 @@ class YoloDataset(Dataset):
     def _update_image_size(self, idx: int) -> None:
         """Update image size based on dynamic shape and batch settings."""
         batch_start_idx = (idx // self.batch_size) * self.batch_size
-        image_ratio = self.ratios[batch_start_idx]
-
+        image_ratio = self.ratios[batch_start_idx].clip(1 / 3, 3)
         shift = ((self.base_size / 32 * (image_ratio - 1)) // (image_ratio + 1)) * 32
 
         self.image_size = [int(self.base_size + shift), int(self.base_size - shift)]