diff --git a/utils/dataloader.py b/utils/dataloader.py index d8baadc69a703d2f8356ced6c83bc8e40ae4af6e..ed6a1b3e61acf2c0ae8fb12a4ec4e2f169c74459 100644 --- a/utils/dataloader.py +++ b/utils/dataloader.py @@ -39,7 +39,7 @@ class YoloDataset(Dataset): # 调整图片大小 new_ar = w / h * self.rand(1 - jitter, 1 + jitter) / self.rand(1 - jitter, 1 + jitter) - scale = self.rand(.25, 2) + scale = self.rand(.5, 1.5) if new_ar < 1: nh = int(scale * h) nw = int(nh * new_ar)