YOLOv8摆设的4种差别摆设方式推理速度对比:pytorch、onnx、ncnn、tflite ...

打印 上一主题 下一主题

主题 528|帖子 528|积分 1584

1.模型转换

首先,我们将yolov8n.pt转换分别转换成onnx、ncnn、tflite格式模型,供后续使用差别模型摆设使用,进行速度对比测试。转换代码如下:
  1. # 转onnx
  2. yolo export model=yolov8n.pt format=onnx
  3. # 转ncnn
  4. yolo export model=yolov8n.pt format=ncnn
  5. # 转tflite
  6. yolo export model=yolov8n.pt format=tflite
复制代码

2. 模型摆设测试

首先,写一个可以测试摄像头与视频的测试脚本VideoTest.py,方便我们进行差别模型的测试。
2.1 pytorch 摆设方式

基于pytorch框架的原生yolov8.pt摆设测试,测试结果如下,检测视频的速度约为2帧/s。命令如下:
   ❝  python VideoTest.py --model=yolov8n.pt --source=1.mp4 --show=True
  


免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。
回复

使用道具 举报

0 个回复

倒序浏览

快速回复

您需要登录后才可以回帖 登录 or 立即注册

本版积分规则

钜形不锈钢水箱

金牌会员
这个人很懒什么都没写!

标签云

快速回复 返回顶部 返回列表