From c6c5fec1d048c745a2cc868b5ac799d302024fa6 Mon Sep 17 00:00:00 2001
From: hexiangdong2019 <PIpi589632147>
Date: Fri, 30 Jul 2021 23:14:51 +0800
Subject: [PATCH] =?UTF-8?q?debug=20eval=5Fbatch=5Fscales=E5=87=BD=E6=95=B0?=
=?UTF-8?q?=E6=9C=89?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit
---
main.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/main.py b/main.py
index 3ff9183..db06a5e 100644
--- a/main.py
+++ b/main.py
@@ -117,11 +117,12 @@ def eval_batch(eval_net, img_lst, crop_size=513, flip=True):
def eval_batch_scales(eval_net, img_lst, scales,
base_crop_size=513, flip=True):
+ print(scales)
sizes_ = [int((base_crop_size - 1) * sc) + 1 for sc in scales]
- probs_lst = eval_batch(args, eval_net, img_lst, crop_size=sizes_[0], flip=flip)
+ probs_lst = eval_batch(eval_net, img_lst, crop_size=sizes_[0], flip=flip)
print(sizes_)
for crop_size_ in sizes_[1:]:
- probs_lst_tmp = eval_batch(args, eval_net, img_lst, crop_size=crop_size_, flip=flip)
+ probs_lst_tmp = eval_batch(eval_net, img_lst, crop_size=crop_size_, flip=flip)
for pl, _ in enumerate(probs_lst):
probs_lst[pl] += probs_lst_tmp[pl]
@@ -145,7 +146,6 @@ def net_eval():
bi = 0
image_num = 0
for i, line in enumerate(img_lst):
- print(line)
img_path = os.path.join('./datasets/JPEGImages/', line+'.bmp')
msk_path = os.path.join('./datasets/SegmentationClass/', line+'.png')
img_ = cv2.imread(img_path)
--
GitLab