news 2026/6/25 13:51:03

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

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
教你如何将yolov8训练好的文件部署在RDK上

一、通用第一步: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用几分钟,开发板用几十分钟吧

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/6/25 13:50:00

1.3 控制幻觉:调试工具在钱面前失灵

2010年5月6日,美国股市发生了一件让全世界工程师脊背发凉的事。 当天下午2点32分,没有任何征兆,道琼斯工业指数在不到五分钟内暴跌近1000点,蒸发市值超过1万亿美元。一些蓝筹股的价格直接归零——埃森哲的股票从40美元瞬间跌到1美…

作者头像 李华
网站建设 2026/6/25 13:49:26

突破性全覆盖路径规划:如何实现机器人100%无死角区域覆盖

突破性全覆盖路径规划:如何实现机器人100%无死角区域覆盖 【免费下载链接】full_coverage_path_planner Full coverage path planning provides a move_base_flex plugin that can plan a path that will fully cover a given area 项目地址: https://gitcode.com…

作者头像 李华
网站建设 2026/6/25 13:46:37

Mythos模型:AI驱动的自动化漏洞挖掘与攻防闭环实践

1. 这不是一次普通模型发布:Mythos 的真实分量与行业震感你可能已经刷到过“Anthropic 发布 Claude Mythos”这条新闻,标题里带着“Preview”“Gated Release”这类字眼,很容易被当成又一场科技公司的例行发布会。但如果你真这么想&#xff0…

作者头像 李华
网站建设 2026/6/25 13:43:32

抖音自动化发布神器:3步实现视频批量智能发布,效率提升500%

抖音自动化发布神器:3步实现视频批量智能发布,效率提升500% 【免费下载链接】douyin_uplod 抖音自动上传发布视频 项目地址: https://gitcode.com/gh_mirrors/do/douyin_uplod 还在为每天重复的抖音视频发布工作烦恼吗?抖音自动化发布…

作者头像 李华