首页 > 其他分享 >香橙派部署RKNN模型出现重复框选和置信度大于1(真解决办法)

香橙派部署RKNN模型出现重复框选和置信度大于1(真解决办法)

时间:2024-10-25 21:19:25浏览次数:8  
标签:RKNN 置信度 NMS rknn 模型 sigmoid 香橙

在香橙派部署yolov5模型时,需要转化两次模型,一次在PC端转为onnx文件,一次在UbuntuPC上转为rknn文件,详情请看这篇博客:

香橙派Pi5Plus(RK3588)部署yolov5模型(不讲环境搭建,只讲步骤)_rk3588 yolov-CSDN博客icon-default.png?t=O83Ahttps://blog.csdn.net/Z0603013xxx/article/details/142737266

1.出现置信度大于1的情况

在不修改任何东西时,可能会出现置信度大于1的情况,这时候是因为,你转化的代码里面没有sigmoid这个函数,注意sigmoid函数不能同时在转onnx和rknn文件中出现。比如,你在转onnx文件中加入了这句话:

那么你在转rknn模型时就不要加sigmoid函数了。

2.解决重复框(在香橙派上的代码上解决)

(不要在UbuntuPC上转rknn模型上改,因为没有NMS这个参数,出现重复框无所谓,我们在香橙派上通过NMS去掉重复框)

在网上很多解决重复框的问题时,他们都以为是sigmoid函数的问题,其实不然。真正的问题其实是NMS(非极大值抑制)。

非极大值抑制(Non-Maximum Suppression,NMS)是一种在计算机视觉领域中常用的技术,特别是在目标检测任务中。NMS的主要作用是在检测过程中去除多余的边界框,从而提高检测的准确性。NMS通过计算候选框之间的交并比(IoU)来决定是否抑制某个框。如果两个框的IoU高于某个阈值(一般为0~0.5),则认为它们是重叠的,较低分数的框会被抑制。

这个参数在官方默认的代码里是0.45,在一般来说0.45有点大了,可以调小一点比如0.1。

修改这个参数,问题解决!

标签:RKNN,置信度,NMS,rknn,模型,sigmoid,香橙
From: https://blog.csdn.net/Z0603013xxx/article/details/143243835

相关文章

  • 香橙派: 拷贝系统到新sd卡(系统备份)
    本文转载自博主:https://www.cnblogs.com/YaoYing/p/12106583.html使用香橙派或者树莓派时,需要烧录系统到sd卡,我们可以重新烧录一张sd卡,也可以将已经配置好系统的sd卡内容拷贝到新卡中。具体步骤如下:把旧SD卡插入树莓派并启动,利用读卡器把新SD卡插入树莓派USB。查看新SD卡......
  • 香橙派5Pro+虚拟机Ubuntu18.04交叉编译+Opencv4.7
    1.基础工具安装①Ubuntu虚拟机下交叉编译工具链安装香橙派5Pro为Arm64位架构,因此需要安装g++-aarch64-linux-gnu以及gcc-aarch64-linux-gnu(有其它交叉编译器,本文以此编译器为例)更新软件源sudoaptupdatesudoaptupgrate点击查看交叉编译工具链安装apt-cachesearcha......
  • 香橙派Zero3 全志H618设备树插件编译
    基本信息全志H618SDK:香橙派官方提供SDK虚拟机:Ubuntu22.04整体的SDK文件夹名称为:orangepi-build-nextSDK中Linux内核路径:orangepi-build-next/kernel/orange-pi-5.4-sun50iw9编译环境指定首先需要指定交叉编译器:在SDK中的toolchains文件夹中含有很多编译器:对于......
  • 【开源】用香橙派Zero 2W做电视机顶盒
    先前用闲置的香橙派3LTS做了一个电视机顶盒,不需要画电路,加上3D打印的外壳就好了。最近又有闲置的香橙派Zero2W,功耗低又很小巧,忍不住又想拿来做机顶盒。香橙派Zero2W算是个核心板,需要配一个扩展板才能得到RJ-45百兆网口、A型USB2.0接口、红外接收头等接口。官方有扩展板成品出......
  • 把廉价香橙派利用到极致,跑自建 Gitea 和 Hexo 服务
    这篇文章在我的博客同步发布:把廉价香橙派利用到极致,跑自建Gitea和Hexo服务书接上回code-server搭建。code-server因为是原生服务,再加上这枚H618芯片性能相当可以,内存和CPU占用都很小,但是耗电却紧压电源适配器的15W极限,剩余的性能……浪费了?这怎么行?!经过苦思冥想,......
  • RK3588安装rknn-toolkit-lite2
    RK3588安装rknn-toolkit-lite2一、下载rknn-toolkit2包:sudogitclonehttps://gitclone.com/github.com/airockchip/rknn-toolkit2二、安装进入到所在包的位置,然后解压:unziprknn-toolkit2-master.zip进入rknn-toolkit-lite2所在路径:cdrknn-toolkit2-master/rknn-t......
  • Deepface - 仅以超过阈值的置信度显示面部的情绪
    我有这段代码可以检测面部表情,但它会在没有面部表情的地方找到面部表情。所以我想知道是否可以放心地做到这一点。我尝试在谷歌上搜索如何做到这一点,但没有找到任何有用的东西。importcv2fromdeepfaceimportDeepFace#Loadfacecascadeclassifierface_cascade=......
  • 【香橙派系列教程】(七)香橙派下的Python3安装
    【七】香橙派下的Python3安装为接下来的Linux图像识别智能垃圾桶做准备。图像处理使用京东SDK只支持pyhton和Java接口,目的是引入C语言的Python调用,感受大厂做的算法bug此接口是人工智能接口,京东识别模型是通过训练后的模型,精准度取决于训练程度,人工智能范畴在常规嵌入式......
  • 【香橙派系列教程】(四)基于ARM-Linux架构的语音控制刷抖音项目
    【四】基于ARM-Linux架构的语音控制刷抖音项目文章目录【四】基于ARM-Linux架构的语音控制刷抖音项目1.语音模块配置1.创建产品2.引脚配置3.词条定义4.添加控制5.发布版本6.烧录固件2.编程实现语音和开发板通信3.手机接入Linux热拔插1.dmesg命令2.adb调试踩坑问题3.总......
  • YOLOv8n部署到RK3588开发板全流程(pt→onnx→rknn模型转换、板端后处理检测)
    YOLOv8n部署到RK3588开发板全流程文章目录前言一、模型训练二、配置用于pt模型转onnx模型的环境三、pt→onnx模型转换四、配置onnx转rknn模型的虚拟环境五、onnx转rknn模型六、RK3588板端部署前言小白博主,第一次写博客记录自己YOLOv8n部署RK3588开发板的全流程,记......