From 72d7e367000744295683cee021d230f24936a727 Mon Sep 17 00:00:00 2001 From: Bubbliiiing <47347516+bubbliiiing@users.noreply.github.com> Date: Mon, 7 Sep 2020 22:06:59 +0800 Subject: [PATCH] Update dataloader.py --- utils/dataloader.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/dataloader.py b/utils/dataloader.py index d8baadc..ed6a1b3 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) -- GitLab