Skip to content

将模型转换到ONNX

简介

将模型转换到ONNX格式,方便后续部署到黑匣设备上使用


模型转换

功能介绍

  1. 模型类型切换按钮,圆点数字代表当前要转换的模型版本,点击即可切换,请根据自己的模型版本切换对应的数字
  2. 模型大小切换按钮,按钮上的标签代表当前模型的输入尺寸(就是如果你传入的图是1920*1080大小的图,在推理的时候会被缩放到这个大小的意思)
  3. 打开文件按钮,点击即可打开文件选择对话框,您在对话框内选择pt模型文件即可 (也可以直接把文件路径输入到目录框内)
  4. 开始转换按钮,点击即可执行转换

将 ONNX 转换为 RKNN 模型

进入到黑匣开放平台,随便编辑一个项目,进入项目编辑页面 点击右上角更多功能按钮

更多功能

在弹出的面板内进行ONNX转换操作即可

RKNN转换

注意:目前版本 yolov5 模型 转换时必须开启 半精度模式 选项

您需要上传的东西有

  • onnx模型文件
  • 50张量化图片(和批量拖拽进入上传框,或直接将源图片文件压缩为 .zip 文件 后上传)
    1. 模型和图标都是通过一个上传口上传的
    2. 上传完毕后即可点击模型转换按钮进行转换,注意查看转换日志
    3. 转换完成后点击上方的 模型文件名即可下载到本地
    4. 一切尘埃落定之后就可以点击 清空缓存 删除此次任务的冗余文件了