1 Star 0 Fork 4

C_HAO/yolov5s_rknn_deploy

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
贡献代码
同步代码
取消
提示: 由于 Git 不支持空文件夾,创建文件夹后会生成空的 .keep 文件
Loading...
README
yolov5s在终端布署上效果还可以,本项目主要达成以下功能:将yolov5s.pt转成yolov5s.onnx,将yolov5s.onnx转成yolov5s.rknn,将yolov5s.rknn部署到rk3399或其他芯片的板子上。
项目地址如下:
https://gitee.com/Dreamdreams8/yolov5s_rknn_deploy/
1、下载第三方库opencv:
链接:https://pan.baidu.com/s/1CvXOsnHHaZzxcMh_-x5Ffg 提取码:qkr7
将3rdparty_yolov5s_rknn_deploy/rknn_to_deploy_3rdparty/opencv放到rknn_to_deploy/examples/3rdparty/中
2、 安装RKNN-Toolkit 
1. 安装 Python3.6 和 pip3,也可以用conda创建一个虚拟环境
sudo apt-get install python3 python3-dev python3-pip
2. 安装相关依赖
sudo apt-get install libxslt1-dev zlib1g zlib1g-dev libglib2.0-0 libsm6 \
libgl1-mesa-glx libprotobuf-dev gcc
3、安装python相关环境
cd onnx_to_rknn
pip install -r requirements-1.1.0.txt
4、安装rknn-toolkit
进入到3rdparty_yolov5s_rknn_deploy/onnx_to_rknn_3rdparty第三方库中
sudo pip3 install rknn_toolkit2*.whl
5、检查 RKNN-Toolkit 是否安装成功
rk@rk:~/rknn-toolkit2/package$ python3
>>> from rknn.api import RKNN
>>>
3、将yolov5s.pt转成yolov5s.onnx
cd yolov5s-to-onnx
1、安装依赖环境,跟上述环境有重叠的地方,一般不冲突
pip install -r yolov5_requirements.txt
2、将yolov5s的模型放到weights中
3、模型转换
python export.py --weights ./weights/yolov5s.pt --img-size 640 --batch 1 --rknn_mode
如果成果则会在weights中生成yolov5s.onnx
4、将yolov5s.onnx转成yolov5s.rknn
1、将生成的yolov5s.onnx放到onnx_to_rknn/examples/onnx/yolov5s中
cd onnx_to_rknn/examples/onnx/yolov5s
2、模型转换
python test.py
如果成功则会在目录下生成yolov5s.onnx,同时生成结果照片

5、将yolov5s.rknn部署到rk3399或其他芯片的板子上。
进入到rknn_to_deploy/examples/yolov5s目录中
执行
bash build-linux.sh
如果成功则生成install跟build两个文件夹
将install文件放到板子上,执行
./yolov5s_demo
可以看到在model生成了out.jpg
走到这一步,恭喜你已经完成模型的布署。
参考来源:https://github.com/mrwangwg123/my-rknn-yolov5
参考来源:https://github.com/ultralytics/yolov5
参考来源:https://github.com/airockchip/yolov5

空文件

简介

本项目主要达成以下功能:将yolov5s.pt转成yolov5s.onnx,将yolov5s.onnx转成yolov5s.rknn,将yolov5s.rknn部署到rk3399或其他芯片的板子上。 展开 收起
C++
取消

发行版

暂无发行版

贡献者

全部

近期动态

加载更多
不能加载更多了
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
C++
1
https://gitee.com/CHAOOO/yolov5s_rknn_deploy.git
git@gitee.com:CHAOOO/yolov5s_rknn_deploy.git
CHAOOO
yolov5s_rknn_deploy
yolov5s_rknn_deploy
master

搜索帮助