From 88f85458b1910af68c5d23769d437cefe25b75e7 Mon Sep 17 00:00:00 2001
From: deepr <hexiangdong2020@outlook.com>
Date: Fri, 13 Aug 2021 21:47:34 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=AD=A3val=E7=9A=84=E5=8A=A0?=
=?UTF-8?q?=E8=BD=BD=E6=96=B9=E5=BC=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit
---
main.py | 10 ++++++----
1 file changed, 6 insertions(+), 4 deletions(-)
diff --git a/main.py b/main.py
index 2e87cc5..1f8aa2c 100644
--- a/main.py
+++ b/main.py
@@ -159,8 +159,9 @@ def net_eval():
batch_msk_lst.append(msk_)
bi += 1
if bi == 4:
- batch_res = eval_batch_scales(eval_net, batch_img_lst, scales=[1.0],
- base_crop_size=513, flip=True)
+ #batch_res = eval_batch_scales(eval_net, batch_img_lst, scales=[1.0],
+ # base_crop_size=513, flip=True)
+ batch_res = batch_img_lst
for mi in range(4):
hist += cal_hist(batch_msk_lst[mi].flatten(), batch_res[mi].flatten(), 6)
@@ -170,8 +171,9 @@ def net_eval():
print('processed {} images'.format(i + 1))
image_num = i
if bi > 0:
- batch_res = eval_batch_scales(eval_net, batch_img_lst, scales=[1.0],
- base_crop_size=513, flip=True)
+ # batch_res = eval_batch_scales(eval_net, batch_img_lst, scales=[1.0],
+ # base_crop_size=513, flip=True)
+ batch_res = batch_img_lst
for mi in range(bi):
hist += cal_hist(batch_msk_lst[mi].flatten(), batch_res[mi].flatten(), 6)
print('processed {} images'.format(image_num + 1))
--
GitLab