如何在yolov8中验证时计算FPS
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}')