diff --git a/tools/infer_kie.py b/tools/infer_kie.py index 41e0856d467582b3560ecc8ae9f20d36aeda891b..16294e59cc51727f39af77d16255ef4d0f2a1bd8 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))