识别框问题

yolov5s训练效果良好（自定义数据集，口罩识别）

https://github.com/airockchip/yolov5.git
https://blog.csdn.net/m0_51714298/article/details/125916417

2. 转换rknn过程正常无报错（onnx2rknn.py）

3. 部署在rk3588上出现问题（输出目标过多）

处理后的图如下（问题未知，暂未解决）：

2022.8.3修改：

经过各种尝试，最终解决：
原来是在转换模型的时候rknn.config中的mean_values与std_values以前测试的时候删掉了。（血泪教训）

笔记：其中还尝试将onnx输出修改为5维[1,3,20,20,7]，原问题不仅没解决，还增加了新问题

切记：rknn模型的输出有三个，分别是：
1×255×80×80
1×255×40×40
1×255×20×20
其中这里的255是85*3，这里的3是指3个anchor产生的3个box，而这里的85是指5+80=85，其中80是类别数量，每个类别数量对应一个label score,—共80个label score,而5是指box的四个坐标加一个box score.

另外，yolov5-6.0输出有四个，但rknn只需三个，具体修改可参考
https://zhuanlan.zhihu.com/p/448651888
再记录一下做的很有用处的笔记
https://blog.csdn.net/Jormungand123/article/details/124420693?utm_medium=distribute.pc_relevant.none-task-blog-2_defaultbaidujs_baidulandingword~default-1-124420693-blog-123525405.pc_relevant_default&spm=1001.2101.3001.4242.2&utm_relevant_index=4

rknn模型转换问题记录相关推荐

瑞芯微RKNN模型转换和PC端仿真
1 模型转换特别说明:以下内容参考来自rknn官网文档 Rockchip_User_Guide_RKNN_Toolkit2_CN-1.3.0.pdf.如有侵权告知删除. 完整代码放在github上. ...
RKNN模型训练转换部署
1. 背景做了一个手机版本的人流量计数,显然不能最终真实环境实施,因为数据集不够,并且硬件还需要搭载其他设备,所以使用的是安卓工控机,但推理速度非常慢,接近500ms,换一个好的CPU,比如3568 ...
yolov5模型转换（pt=＞onnx=＞rknn）和板端验证测试
测试环境说明: (1)由于模型转换工具需要onnx版本和rknn的tool工具需要的版本相互矛盾需要创建量开发环境,当前测试转换的模型是yolov5_v5.0的模型 (2)由于在搭建开发环境时还存在部 ...
rknn3399pro 2小时入门指南（三）在PC上搭建RKNN模拟环境、模型转换代码、RK3399 pro使用
目录 1.前言官方论坛说明 2.PC-ubuntu系统下tensorflow+keras虚拟环境搭建 3.在PC-ubuntu下搭建rknn模拟环境搭建 4.相关转化代码 (1) h5 -> ...
yolov5模型转换rknn
yolov5模型转换rknn 安装virtualenv,用来来管理python环境导出包到另一个系统图片标注安装yolo onnx模型转换为rknn RK3588安装RKNN-toolkit 安 ...
原理 | 分布式链路跟踪组件 SOFATracer 和 Zipkin 模型转换
SOFA 中间件是蚂蚁金服自主研发的金融级分布式中间件,包含了构建金融级云原生架构所需的各个组件,包括微服务研发框架,RPC 框架,服务注册中心,分布式定时任务,限流/熔断框架,动态配置推送,分布式链 ...
模型转换、压缩、加速工具
20210618 sky_hole: 回成都工作了吗?wang shi yang: 嗯我现在已经在成都上班了sky_hole: 不用付费,我之前发你的视频你好好看看就可以入门了sky_hole: 成 ...
【地平线开发板模型转换】将pytorch生成的onnx模型转换成.bin模型
文章目录 1 获取onnx模型 2 启动docker容器 3 onnx模型检查 3.1 为什么要检查? 3.2 如何操作 4 图像数据预处理 4.1 一些问题的思考 4.2 图片挑选与放置 4.2 使 ...
瑞芯微RV1126部署yolov5-face_模型转换_输出后处理C++实现
目录 1.模型转换 1.1 yolov5-face的pt模型转为onnx模型 1.2 yolov5-face的onnx模型转换为rknn模型 2.C++代码 2.1 paddi
模型转换、模型压缩、模型加速工具汇总
点击上方"计算机视觉工坊",选择"星标" 干货第一时间送达编辑丨机器学习AI算法工程一.场景需求解读在现实场景中,我们经常会遇到这样一个问题,即某篇论 ...

rknn模型转换问题记录

rknn模型转换问题记录

识别框问题

2022.8.3修改：

rknn模型转换问题记录相关推荐

最新文章

热门文章