YOLOv5 YOLOv7 YOLOv8的模型测试对比和结论
背景说明
基于使用自己训练的模型,yolov8做推理测试的时候postprocess时间大于inference时间_故做此研究。在val和predict速度栏,刚开始发现Yolov8的postprocess时长达到了16ms+。【昂。。。好尴尬,最后发现预测和验证的时候,nvidia-smi显示的显卡信息被另外一个python程序占用了20G+的量,后面没有其它python程序的时候,显示Speed结果如表2 各模型训练结果对比所示,猜测可能做nms的时候,CPU被占用较多,所以影响了postprocess的速率。实际nms的速度远小于inference的速度。】
自己用9285:1985:1987做了训练和验证测试。
(1)训练平台信息
表1 训练平台相关信息
系统: | Ubuntu20.04 | |
---|---|---|
Driver Version: | 525.78.01 |