FPS(Frames Per Second),每秒多少帧
在 val.py
中,需要修改参数 batch-size=1
同时别忘了修改自己数据集的 yaml 文件以及训练好的 pt 文件,如下图:
修改完成后,运行 val.py
,得到结果,如下图:
上图中显示的三个时间(ms)——pre-process
:图像预处理, inference
:推理, NMS
: 非极大值抑制,这三个时间相加就是模型处理一张图像所需的时间。
将结果 四舍五入 为最接近的整数,便是最终求得的 FPS
另外,要想在控制台打印 FPS 可以在 val.py
中按住快捷键 Ctrl+F
搜索 Print speeds
定位过去,然后添加如下代码:
FPS = 1000 / sum(t)
LOGGER.info(f'FPS: {round(FPS, 0)}')
如下图: