ultralytics-main/ultralytics/engine/validator.py文件下,第200行左右,添加如下代码

LOGGER.info(f'FPS:{(1000 / sum(self.speed.values())):.2f}')

       speed.values()是一个字典,包括preprocess,inference,loss,postprocess的时间,所以这行代码计算的fps包括对图片前处理,推理和后处理的总时间,每次运行可能FPS会有波动,如果想要一个稳定的数值,可以运行多次求平均。

        如果只想要纯网络的推理时间,可以使用如下代码

LOGGER.info(f'FPS:{(1000 / tuple(self.speed.values())[1]):.2f}')
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐