实例讲解昇腾 CANN YOLOV8 和 YOLOV9 适配
💡
原文中文,约13600字,阅读约需33分钟。
📝
内容提要
本文介绍了华为昇腾CANN YOLOV8和YOLOV9的适配过程。首先需要获取YOLOV8的模型文件,并将其转化为.onnx模型。然后在Atlas 500 Pro服务器上进行模型转换。适配代码可以参考开源代码。最后进行编译运行,输出结果为检测到的物体和置信度。适配YOLOV9的过程与YOLOV8类似。
🎯
关键要点
- 本文介绍了华为昇腾CANN YOLOV8和YOLOV9的适配过程。
- 适配YOLOV8需要获取模型文件并转化为.onnx模型。
- 在Atlas 500 Pro服务器上进行模型转换,适配代码可参考开源代码。
- YOLOV8模型输出数据大小为[1,84,8400],需要修改后处理部分以适配YOLOV8/YOLOV9。
- 在Windows上安装YOLOV8环境并执行python脚本将.pt模型转化为.onnx模型。
- 上传yolov8n.onnx模型到Atlas 500 Pro服务器并执行模型转换命令。
- 适配代码已开源,核心代码在sampleYOLOV8.cpp中。
- 编译运行代码后,生成的可执行文件可以进行物体检测。
- YOLOV9的适配过程与YOLOV8类似,输出格式一致,只需生成yolov9xx.om模型。
- YOLOV9模型转换步骤包括创建新环境、克隆代码、安装依赖和导出onnx模型。
🏷️
标签
➡️