add ips

2021-12-19 07:00:00 +00:00 · 2021-12-19 07:00:00 +00:00 · 1c7f93060c
parent 9dd50e6a90
commit 1c7f93060c
1 changed files with 1 additions and 1 deletions
--- 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))