设为首页
收藏本站
开启辅助访问
切换到窄版
登录
立即注册
只需一步,快速开始
首页
BBS
剑灵怀旧服
Unity
美术资源
UE
全套源码
其他内容
奖励任务
分享
Share
每日任务中心
搜索
搜索
怀旧插件
怀旧脚本数据
怀旧攻略
公益服开服通知
AI工具分享
Unity插件
Unity问题答疑
Unity学习教程
Unity AI算法
Unity源码
其他美术文件
MAY文件
3D Max
FBX模型
UE技术讨论
UE学习视频
UE源码
网站源代码
手机游戏源代码
PC游戏
游戏捏脸
其他插件
OFFER摸吧
魔兽地图
单机游戏
本版
帖子
用户
爱开源网
»
首页
›
数字人
›
人工智能
›
yolov5篇---yolov5练习pt模子并转换为rknn模子,摆设在R ...
返回列表
发新帖
yolov5篇---yolov5练习pt模子并转换为rknn模子,摆设在RK3588开发板上——
[复制链接]
509
|
0
|
2023-8-23 11:40:04
|
显示全部楼层
|
阅读模式
一、任务先容
瑞芯微RK3588是一款搭载了NPU的国产开发板。
NPU
(neural-network processing units)可以说是为了嵌入式神经网络和边缘计算量身定制的,但若想调用RK3588的NPU单元举行推理加速,则须要首先将模子转换为**.rknn格式**的模子,否则无法利用。
这次我们的任务是将yolov5练习得到的
pt模子
,一步步转换为rknn模子,并将rknn模子摆设在搭载RK3588的StationPC M3主机上,利用NPU推理。(官网NPU教程:NPU利用 — Firefly Wiki)查阅资料和官网后发现转换和利用过程分以下几步,首先总结给大家,后文细说:
1.利用
准确版本(v5.0)
的yolov5举行练习得到
pt模子
;
2.将pt模子利用yolov5工程中的export.py转换为
onnx模子
;
3.将
onnx
模子利用
rknn-toolkit2
中onnx文件夹的test.py转换为rknn模子;
4.在板子上利用
rknpu2
工具调用rknn模子,实现NPU推理加速。
接下来举行具体先容。
二、实行过程
2.1 利用准确版本的yolov5举行练习(平台:x86机器ubuntu22.04体系)
人菜就要跟着官方教程做,请看官方教程:
rknn-toolkit/examples/pytorch/yolov5 at master · rockchip-linux/rknn-toolkit · GitHub
须要指出的是,固然我们后面要利用的是rknn-toolkit2工具举行模子转换,但教程却在rknn-toolkit工程中,github二者的父目次如图:
进入rknn-toolkit工程,欣赏至
/example/pytorch/yolov5
,在README中赫然写着:
这就是我们本次要利用的“
准确版本的yolov5
” ,怕贫困的读者请肯定按照这个id去yolov5官网找到工程并克隆下来。进入yolov5官网GitHub - ultralytics/yolov5: YOLOv5
本帖子中包含更多资源
您需要
登录
才可以下载或查看,没有账号?
立即注册
x
回复
使用道具
举报
返回列表
发新帖
高级模式
B
Color
Image
Link
Quote
Code
Smilies
您需要登录后才可以回帖
登录
|
立即注册
本版积分规则
发表回复
回帖后跳转到最后一页
盗贼
17
主题
0
回帖
51
积分
注册会员
注册会员, 积分 51, 距离下一级还需 299 积分
注册会员, 积分 51, 距离下一级还需 299 积分
积分
51
加好友
发消息
回复楼主
返回列表
Omniverse
人工智能
其他
图文推荐
非绿色-黑鳍自动换角色组队吃药自动没疲劳刷金
2024-09-21
绿色-剑灵怀旧完整主线(10月更新)有韩服
2024-09-12
非绿色-定制AI全自动黑鳍V1.2
2024-09-14
剑灵三系精修端 星术咒3鬼3 T2 饰品属性外观
2023-08-16
Unity 超级马里奥
2023-08-22
热门排行
1
非绿色-黑鳍自动换角色组队吃药自动没疲劳
2
非绿色-定制AI全自动黑鳍V1.2
3
绿色-剑灵怀旧完整主线(10月更新)有韩服
4
剑灵三系精修端 星术咒3鬼3 T2 饰品属性外
5
Unity 超级马里奥
6
AI绘画stable-diffusion-webui指定GPU运行
7
非绿色-武神塔合集
8
非绿色-使用教程攻略
9
非绿色 3号自动切线 黑骑 钓鱼 蓝蛙
10
绿色-黑鳍脚本合集