From f724489e3e99b2d7a78069a3225ba88a95aa19e2 Mon Sep 17 00:00:00 2001 From: guuske31 <50746212+guuske31@users.noreply.github.com> Date: Thu, 23 May 2019 12:54:08 +0200 Subject: [PATCH] Update utils.py --- core/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/core/utils.py b/core/utils.py index 27e19388b..587079330 100644 --- a/core/utils.py +++ b/core/utils.py @@ -37,8 +37,8 @@ def get_anchors(anchors_path): def image_preporcess(image, target_size, gt_boxes=None): - image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB).astype(np.float32) - + # image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB).astype(np.float32) + image = (np.float32(image), cv2.COLOR_BGR2RGB)#.astype(np.float32) ih, iw = target_size h, w, _ = image.shape