提交 1c7f9306 编写于 作者: L LDOUBLEV

add ips

上级 9dd50e6a
......@@ -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))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册