yoloV8导出engine模型

yoloV8导出engine模型

引言:
目的是TensorRT加速YOLO实例分割模型。
本博客记录达成此目的所需的步骤,及步骤中可能遇到的问题。
as follow:

1. 首先导出onnx模型的脚本命令:

yolo export model=best.pt format=onnx opset=12 simplify=True

脚本命令的参数:
yolo导出模型参数
不同格式使用的模型对应:
在这里插入图片描述
我遇到的几个问题:

ImportError: DLL load failed while importing _imaging: 找不到指定的模块。
解决:pip卸载Pillow,再重新安装

pip uninstall Pillow
pip install Pillow

ModuleNotFoundError: No module named 'onnx'
解决:pip安装onnx

pip install onnx

requirements: Ultralytics requirements ['cmake', 'onnxsim>=0.4.33', 'onnxruntime'] not found, attempting AutoUpdate...
requirements: ❌ AutoUpdate skipped (offline)
解决:pip安装cmake、onnxsim【版本>=0.4.33,可能遇到文件名过长的问题】、onnxruntime
在这里插入图片描述

pip install cmake

pip install onnxsim==0.4.33 #可能会报错文件名过长【原因可能是安装python的时候没有禁用文件名长度限制】
#进入conda虚拟环境后
mountvol -x D:\ /s
#然后再次
pip install onnxsim==0.4.33

pip install onnxruntime

最后导出best.onnx模型:
在这里插入图片描述

2. 先测试一下导出onnx模型【脚本命令】:

yolo task=segment mode=predict model=best.onnx source=18.JPG show=True save=True device=0

在这里插入图片描述
测试效果:
在这里插入图片描述

3. TensorRT安装、onnx转engine:

TensorRT官网:https://developer.nvidia.com/tensorrt/download
在这里插入图片描述
在这里插入图片描述
安装教程【参考博客】:https://blog.csdn.net/qq_44747572/article/details/129022225
版本对应【参考博客】:
https://blog.csdn.net/initative/article/details/138213992

安装完成后,可以使用trtexec 命令:

trtexec --onnx=best.onnx --saveEngine=best.engine

即可得到yolo的engine模型。
trtexec命令参数【参考博客】:
https://blog.csdn.net/HW140701/article/details/120360642

相关推荐

  1. yolov8模型转换

    2024-07-13 14:46:04       44 阅读
  2. YOLOv8模型部署

    2024-07-13 14:46:04       54 阅读
  3. YOLOV5 TensorRT部署 BatchedNMS(转换engine模型)(上)

    2024-07-13 14:46:04       32 阅读
  4. yolov5导出onnx模型问题

    2024-07-13 14:46:04       45 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-07-13 14:46:04       67 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-07-13 14:46:04       71 阅读
  3. 在Django里面运行非项目文件

    2024-07-13 14:46:04       58 阅读
  4. Python语言-面向对象

    2024-07-13 14:46:04       69 阅读

热门阅读

  1. go-基准测试

    2024-07-13 14:46:04       24 阅读
  2. kafka部署以及常用命令详细总结

    2024-07-13 14:46:04       18 阅读
  3. windows安全加固

    2024-07-13 14:46:04       18 阅读
  4. 微服务架构实战:案例分析与解决方案探讨

    2024-07-13 14:46:04       22 阅读
  5. 【数据结构】B树

    2024-07-13 14:46:04       18 阅读
  6. kafka 常用命令

    2024-07-13 14:46:04       21 阅读
  7. 解压zip、tar、gz文件--linux003

    2024-07-13 14:46:04       16 阅读
  8. Python调用摄像头

    2024-07-13 14:46:04       20 阅读
  9. [AI 大模型] OpenAI ChatGPT

    2024-07-13 14:46:04       22 阅读