如何在yolov8中验证时计算FPS

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

如何在yolov8中验证时计算FPS_第1张图片

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}')

你可能感兴趣的:(YOLO)