news 2026/5/1 8:42:22

解析如何使用官方的YOLOv11代码pt权重转换出适配瑞芯微ONNX➡RKNN的权重文件

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
解析如何使用官方的YOLOv11代码pt权重转换出适配瑞芯微ONNX➡RKNN的权重文件

前言

在ultralytics中,导出的yolo11.onnx权重只有一个输出,这无法适配瑞芯微中的解析。其需要九个输出。通用的步骤是在训练完的pt文件后,利用[ultralytics_yolo11],可参考:【YOLOv8部署至RV1126】PT转ONNX、ONNX转RKNN、RKNN预编译

官方导出的onnx权重的输出个数:


利用ultralytics_yolo11导出的onnx权重的输出个数:

为了进一步理解其是如何在官方代码的基础上将其原本一个输出改成九个输出,本文主要使用官方的ultralytics代码(8.3.13版本),将yolo11n.pt文件转成onnx➡rknn➡预编译,比较一下两种转换后的权重是否一致。

文章目录

  • 前言
  • 一、修改ultralytics
  • 二、导出ONNX
  • 三、导出RKNN
  • 四、预编译

一、修改ultralytics

在ultralytics/nn/modules/head.py中的Detect类的forward函数中添加如下代码:

# 导出适配瑞芯微的ONNX模型ifself.export:y=[]foriinrange(self.nl):
版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/4/24 14:26:12

Z-Image-Turbo启动报错?supervisorctl start命令执行失败排查教程

Z-Image-Turbo启动报错?supervisorctl start命令执行失败排查教程 1. 引言 1.1 业务场景描述 Z-Image-Turbo 是阿里巴巴通义实验室开源的高效AI图像生成模型,作为 Z-Image 的知识蒸馏版本,它在保持高质量图像输出的同时大幅提升了推理速度…

作者头像 李华
网站建设 2026/4/18 9:42:36

求职简历照片烦恼终结者!AI工坊实战应用案例分享

求职简历照片烦恼终结者!AI工坊实战应用案例分享 1. 引言:证件照制作的现实痛点与AI破局 在求职、考证、签证等场景中,一张合规、清晰、专业的证件照是不可或缺的“第一印象”。然而,传统获取方式存在诸多不便: 前往…

作者头像 李华
网站建设 2026/4/30 14:09:51

Qwen3-Reranker-0.6B实战案例:云端10分钟上手,2块钱低成本验证

Qwen3-Reranker-0.6B实战案例:云端10分钟上手,2块钱低成本验证 你是不是也遇到过这样的情况?作为产品经理,看到竞品在搜索结果排序、推荐系统或问答匹配上用了“重排序”技术,用户体验明显提升,心里也开始…

作者头像 李华
网站建设 2026/5/1 8:39:22

Qwen-Image-2512-ComfyUI部署要点:依赖库安装与版本匹配

Qwen-Image-2512-ComfyUI部署要点:依赖库安装与版本匹配 1. 引言 1.1 业务场景描述 随着多模态生成模型的快速发展,图像生成技术在设计、内容创作和AI艺术领域展现出巨大潜力。阿里云推出的Qwen-Image-2512作为其通义千问系列中的最新图像生成模型&am…

作者头像 李华
网站建设 2026/4/27 18:28:31

HunyuanVideo-Foley恐怖氛围:阴森背景音与突发惊吓音效设计

HunyuanVideo-Foley恐怖氛围:阴森背景音与突发惊吓音效设计 1. 技术背景与应用场景 随着AI生成技术在多媒体领域的深入发展,音效自动生成正成为视频制作流程中不可或缺的一环。传统音效设计依赖专业音频工程师手动匹配动作与声音,耗时且成本…

作者头像 李华
网站建设 2026/5/1 7:23:39

Wan2.2部署优化:小显存GPU运行50亿参数模型的实战经验分享

Wan2.2部署优化:小显存GPU运行50亿参数模型的实战经验分享 近年来,文本到视频(Text-to-Video)生成技术迅速发展,成为AIGC领域的重要方向。然而,大多数高质量视频生成模型对计算资源要求极高,尤…

作者头像 李华