Merge pull request #1156 from will-jl944/bug_fix

fix timer bug

paddlex/deploy.py

@@ -253,6 +253,7 @@ class Predictor(object):
                 images=images, topk=topk, transforms=transforms)
 
         self.timer.repeats = repeats
+        self.timer.img_num = len(images)
         self.timer.info(average=True)
 
         return results

paddlex/utils/utils.py

@@ -179,11 +179,12 @@ class Timer(Times):
 
     def info(self, average=False):
         total_time = self.preprocess_time_s.value(
-        ) + self.inference_time_s.value() + self.postprocess_time_s.value()
+        ) * self.img_num + self.inference_time_s.value(
+        ) + self.postprocess_time_s.value() * self.img_num
         total_time = round(total_time, 4)
         print("------------------ Inference Time Info ----------------------")
         print("total_time(ms): {}, img_num: {}, batch_size: {}".format(
-            total_time * 1000, self.img_num, self.img_num / self.repeats))
+            total_time * 1000, self.img_num, self.img_num))
         preprocess_time = round(
             self.preprocess_time_s.value() / self.repeats,
             4) if average else self.preprocess_time_s.value()