add ips

1c7f9306 · LDOUBLEV · 9dd50e6a · 1c7f9306
Commit 1c7f9306 authored Dec 19, 2021 by LDOUBLEV
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tools/infer_kie.py tools/infer_kie.py +1 -1

No files found.
--- 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))