From 1c7f93060cf4cbfdfca5669a4d4d502c531cfff3 Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Sun, 19 Dec 2021 07:00:00 +0000 Subject: [PATCH] add ips --- tools/infer_kie.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/infer_kie.py b/tools/infer_kie.py index 41e0856d..16294e59 100755 --- a/tools/infer_kie.py +++ b/tools/infer_kie.py @@ -144,7 +144,7 @@ def main(): logger.info("success!") logger.info("It took {} s for predict {} images.".format( np.sum(count_t), len(count_t))) - ips = np.sum(count_t[warmup_times:]) / len(count_t[warmup_times:]) + ips = len(count_t[warmup_times:]) / np.sum(count_t[warmup_times:]) logger.info("The ips is {} images/s".format(ips)) -- GitLab