YOLO v5 FPS计算方法_yolov5中fps如何根据参数量和运算量计算

作者：AllinToyou | 2024-04-12 19:01:35

踩

yolov5中fps如何根据参数量和运算量计算

相信大家也遇到了和我一样的问题，FPS究竟该怎么算，本人今天澄清一下。因为网上各种说法也没说清楚

在val.py中，需要修改参数batch-size=1!!!!!!!!切记，否则算下来就是错的

其次运行脚本python val.py --data data/voc_bm.yaml --weights runs/train/yolov5s/weights/best.pt --batch-size 1（当然也可以命令行参数指定为1，其他参数可以根据自己需要调整）

运行之后得到如下图

$FPS=1000ms/(0.3+12.0+1.4)$

pre-process：图像预处理时间，包括图像保持长宽比缩放和padding填充，通道变换（HWC->CHW）和升维处理等；
inference：推理速度，指预处理之后的图像输入模型到模型输出结果的时间；
NMS ：你可以理解为后处理时间，对模型输出结果经行转换等；

大家明白了吗？

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/AllinToyou/article/detail/412186