From 4ba8fae0cfd59fbd77c73d49c5144c9216cabfe8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=BE=90=E6=97=A6=E9=BA=92?= Date: Tue, 19 May 2020 16:54:11 +0800 Subject: [PATCH] fix bug --- core/dataset.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/core/dataset.py b/core/dataset.py index 485e21e79..1df3fbd99 100644 --- a/core/dataset.py +++ b/core/dataset.py @@ -119,8 +119,8 @@ def random_crop(self, image, bboxes): crop_xmin = max(0, int(max_bbox[0] - random.uniform(0, max_l_trans))) crop_ymin = max(0, int(max_bbox[1] - random.uniform(0, max_u_trans))) - crop_xmax = max(w, int(max_bbox[2] + random.uniform(0, max_r_trans))) - crop_ymax = max(h, int(max_bbox[3] + random.uniform(0, max_d_trans))) + crop_xmax = min(w, int(max_bbox[2] + random.uniform(0, max_r_trans))) + crop_ymax = min(h, int(max_bbox[3] + random.uniform(0, max_d_trans))) image = image[crop_ymin : crop_ymax, crop_xmin : crop_xmax]