From 951064db80955456fa1aa0064969d819d740aa97 Mon Sep 17 00:00:00 2001
From: deepr <hexiangdong2020@outlook.com>
Date: Wed, 11 Aug 2021 15:33:24 +0800
Subject: [PATCH] =?UTF-8?q?=E5=8F=96=E6=B6=88preprocess?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit
---
main.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/main.py b/main.py
index 2761769..7a7f0e5 100644
--- a/main.py
+++ b/main.py
@@ -95,7 +95,8 @@ def eval_batch(eval_net, img_lst, crop_size=513, flip=True):
resize_hw = []
for l in range(batch_size):
img_ = img_lst[l]
- img_, resize_h, resize_w = pre_process(img_, crop_size)
+ resize_h, resize_w, _ = img_.shape
+ # img_, resize_h, resize_w = pre_process(img_, crop_size)
batch_img[l] = img_
resize_hw.append([resize_h, resize_w])
--
GitLab