YOLOv5 如何计算并打印 FPS

文章用于学习记录

• YOLO v5 FPS计算方法
• 修改对应自己数据集的 yaml 文件以及训练好的 pt 文件以及batch-size=1,
  

  FPS = 1000ms/(0.3+11.9+1.0)
  

• pre-process：图像预处理时间，包括图像保持长宽比缩放和padding填充，通道变换（HWC->CHW）和升维处理等；
• inference：推理速度，指预处理之后的图像输入模型到模型输出结果的时间；
• NMS ：非极大值抑制处理，可以理解为后处理时间，对模型输出结果经行转换等；
• YOLO v5 FPS打印实现步骤：
  • 在 val.py 中按住快捷键 Ctrl+F 搜索 Print speeds 定位过去
    
  • 在上图中的合适位置添加如下代码：

    FPS = 1000 / sum(t)
    LOGGER.info(f'FPS: {round(FPS, 3)}')