未验证 提交 651d9ce6 编写于 作者: Z zhiboniu 提交者: GitHub

fix pose deploy time info bug:division by zero (#3832)

上级 86760aed
...@@ -170,15 +170,16 @@ class Timer(Times): ...@@ -170,15 +170,16 @@ class Timer(Times):
print("total_time(ms): {}, img_num: {}".format(total_time * 1000, print("total_time(ms): {}, img_num: {}".format(total_time * 1000,
self.img_num)) self.img_num))
preprocess_time = round( preprocess_time = round(
self.preprocess_time_s.value() / self.img_num, self.preprocess_time_s.value() / max(1, self.img_num),
4) if average else self.preprocess_time_s.value() 4) if average else self.preprocess_time_s.value()
postprocess_time = round( postprocess_time = round(
self.postprocess_time_s.value() / self.img_num, self.postprocess_time_s.value() / max(1, self.img_num),
4) if average else self.postprocess_time_s.value() 4) if average else self.postprocess_time_s.value()
inference_time = round(self.inference_time_s.value() / self.img_num, inference_time = round(self.inference_time_s.value() /
max(1, self.img_num),
4) if average else self.inference_time_s.value() 4) if average else self.inference_time_s.value()
average_latency = total_time / self.img_num average_latency = total_time / max(1, self.img_num)
print("average latency time(ms): {:.2f}, QPS: {:2f}".format( print("average latency time(ms): {:.2f}, QPS: {:2f}".format(
average_latency * 1000, 1 / average_latency)) average_latency * 1000, 1 / average_latency))
print( print(
...@@ -189,13 +190,13 @@ class Timer(Times): ...@@ -189,13 +190,13 @@ class Timer(Times):
def report(self, average=False): def report(self, average=False):
dic = {} dic = {}
dic['preprocess_time_s'] = round( dic['preprocess_time_s'] = round(
self.preprocess_time_s.value() / self.img_num, self.preprocess_time_s.value() / max(1, self.img_num),
4) if average else self.preprocess_time_s.value() 4) if average else self.preprocess_time_s.value()
dic['postprocess_time_s'] = round( dic['postprocess_time_s'] = round(
self.postprocess_time_s.value() / self.img_num, self.postprocess_time_s.value() / max(1, self.img_num),
4) if average else self.postprocess_time_s.value() 4) if average else self.postprocess_time_s.value()
dic['inference_time_s'] = round( dic['inference_time_s'] = round(
self.inference_time_s.value() / self.img_num, self.inference_time_s.value() / max(1, self.img_num),
4) if average else self.inference_time_s.value() 4) if average else self.inference_time_s.value()
dic['img_num'] = self.img_num dic['img_num'] = self.img_num
total_time = self.preprocess_time_s.value( total_time = self.preprocess_time_s.value(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册