🔀 [Merge] branch 'DATASET' into TEST

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)]
@@ -214,7 +213,7 @@ def create_dataloader(data_cfg: DataConfig, dataset_cfg: DatasetConfig, task: st
     if task == "inference":
         return StreamDataLoader(data_cfg)
 
-    if dataset_cfg.auto_download:
+    if getattr(dataset_cfg, "auto_download", False):
         prepare_dataset(dataset_cfg, task)
     dataset = YoloDataset(data_cfg, dataset_cfg, task)