尧图网站建设 尧图网络
  • 首页
  • 关于我们
  • 服务项目
  • 案例展示
  • 建站流程
  • 资讯中心
  • 联系我们
首页/资讯中心/详情

教你如何将yolov8训练好的文件部署在RDK上

教你如何将yolov8训练好的文件部署在RDK上
📅 发布时间:2026/6/25 14:07:04

一、通用第一步:PC 端把.pt 导出 ONNX

因为我的训练文件是.pt文件

先装好 ultralytics,没装先执行:pip install ultralytics onnx onnxsim

这一步你可能会出错,因为你没装cmake,编译安装 onnxsim 需要cmake工具

如果出错就执行pip install cmake然后pip install onnxsim就成功了

再通过你的终端cd到你训练文件所在的文件夹比如我的:

cd C:\Users\123\ultralytics-main\runs\detect\train-3\weights

然后运行一个命令:yolo export model=best.pt format=onnx imgsz=640 simplify=True opset=17 dynamic=False

会自动生成一个onnx文件,然后再次精简 ONNX(必做,解决后续量化报错)

当前目录已经有best.onnx,执行简化:会生成一个best_simplified.onnx

二、准备校准图片,开始量化

  • 从你之前训练用的图片里,随便复制 100~200 张原图,新建文件夹calib

  • 新建dataset.txt文本,每一行粘贴一张图片的完整路径 示例:

  • 示例:C:\train\calib\img1.jpg C:\train\calib\img2.jpg把三样东西放同一个文件夹:best_simplified.onnx+ calib 图片文件夹 + dataset.txt

  • 对于粘贴路径你可以在Windows照片文件夹里上一位文件夹里新建文本,改名gen_txt.bat写入下面代码:@echo off dir /s /b calib\*.jpg > dataset_win.txt双击运行这个 bat,自动生成dataset_win.txt,里面全部是图片完整 Windows 路径。

  • 把 Windows 路径转成 WSL 识别路径

    打开 dataset_win.txt,Ctrl+H 全局替换:

  • 查找:C:\替换为:/mnt/c/

  • 查找:\替换为:/保存为dataset.txt,直接给 Docker 用,几百张图 10 秒搞定

  • 安装 Docker,导入地平线 X5 工具镜像,启动容器,把上面存放模型、校准图的本地文件夹映射进容器 sudo docker run -it --rm --gpus all \

  • -v /本地文件夹路径:/open_explorer/user_data \ openexplorer/ai_toolchain_ubuntu_20_x5_gpu:v1.2.8-py310

  • 在容器内写转换脚本,填入 onnx 路径、图片文件、归一化参数、硬件型号 X5

  • 运行脚本自动完成两件事:

    • 遍历 dataset.txt 里所有图片,输入模型统计每层数值范围(校准)

    • 根据统计数据把浮点模型转为 INT8 整数模型,输出yolo8.bin作用:适配 RDK X5 芯片,大幅提升运行速度,降低功耗

    • 最后将.bin文件拷到RDK上就行

  • 当然如果你不想在WSL或者ubuntu中量化你也可以选择直接在RDK上量化只是因为开发板性能比较弱,WSL/ubuntu用几分钟,开发板用几十分钟吧

相关新闻

  • 面试官问:“你怎么评估一个 Agent 到底好不好用?”,我笑了:“试了几个问题,没问题就行”,面试官:“你不叫评估,叫碰运气”
  • 终极指南:如何用DroneSecurity深度解析DJI无人机通信协议?
  • 临床AI落地五大生死线:从模型可信度到人机协同的实战指南

最新新闻

  • 专业EMT全套检测试剂|云克隆一站式上皮间质转化研究解决方案
  • Python机器学习入门实战:线性回归、决策树、KNN、朴素贝叶斯四算法手把手实现
  • 2026年【江苏“信息与未来”编程思维】真题及题解(T3:南京名片)
  • AI 写小说长篇记忆技术深度研究报告
  • UI Recorder架构解析:Chrome扩展与Node.js如何协同实现自动化测试
  • C语言:模块化开发与Makefile精讲

日新闻

  • 利用微PE工具箱进行系统安装教程
  • 渗透测试十大核心工具实战指南:从信息搜集到报告生成全流程解析
  • 暗黑破坏神2存档编辑器:网页版角色修改工具完全指南

周新闻

  • Visual C++运行库修复终极指南:5分钟快速解决Windows软件启动错误
  • 手把手教你构建统计局地区经济数据爬虫:从环境搭建到数据持久化全指南
  • 2026多Agent深度解析:用AI团队替代单一模型,四种架构实战落地

月新闻

  • 【总结】入门篇:50句话让你记住架构核心概念
  • WeChatMsg技术方案解析:实现Mac微信数据自主管理的完整解决方案
  • WeChatMsg:革新性微信数据备份方案,打造你的专属数字记忆库

关于尧图

  • 公司简介
  • 团队介绍
  • 企业文化
  • 荣誉资质

服务项目

  • 定制开发
  • 电商建站
  • UI 设计
  • 运维服务

快速链接

  • 案例展示
  • 建站流程
  • 常见问题
  • 资讯中心

联系方式

  • 📍北京市朝阳区互联网产业园 A 座 10 层
  • 📞400-888-8888
  • ✉️contact@rkmt.cn
  • 🕐周一至周日 9:00-21:00

© 2024 北京尧图网络科技有限公司 版权所有 | 京 ICP 备 XXXXXXXX 号