From b9ad4bfcd8858ee3359dad594f565fd9f1eb449c Mon Sep 17 00:00:00 2001
From: deepr <hexiangdong2020@outlook.com>
Date: Fri, 13 Aug 2021 22:05:13 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9F=A5=E7=9C=8Bimg=5F=E7=9A=84shape?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit
---
main.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/main.py b/main.py
index 1a20473..4339cf1 100644
--- a/main.py
+++ b/main.py
@@ -130,6 +130,7 @@ def eval_batch_scales(eval_net, img_lst, scales,
sizes_ = [int((base_crop_size - 1) * sc) + 1 for sc in scales]
probs_lst = eval_batch(eval_net, img_lst, crop_size=sizes_[0], flip=flip)
for crop_size_ in sizes_[1:]:
+ print(img_lst.shape)
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]
--
GitLab