- 博客(6)
- 收藏
- 关注
原创 yolov8训练个人数据集 + onnxruntime/onnxruntime-gpu封装加速部署(Windows超详细教程)
本文详细介绍了YOLOv8模型从训练到ONNX部署的完整流程。主要内容包括:1)环境配置(CUDA、cuDNN、ONNX Runtime);2)YOLOv8模型训练步骤(数据集准备、配置文件修改、训练过程);3)模型转换(PT转ONNX);4)ONNX推理实现(前处理、推理、后处理)。重点说明了ONNX Runtime的GPU加速使用方法,并提供了完整的推理类代码实现,包含图像预处理、推理和后处理(NMS)的全流程。教程适用于Windows系统,同时兼容CPU推理场景。
2025-06-15 10:12:30
722
原创 pyside6开发篇---隐藏标题栏后,实现关闭、最大(小)化、窗口拖动、窗口拖动调整大小
代码很简单,但是这样就会失去最大化、窗口拖动、窗口调整大小等功能,需要我们手动实现。
2024-07-21 12:58:19
3006
1
原创 Docker---windows下载安装及配置(解决pull时间失败、过长问题)
官网下载地址(最新版):,需要魔法工具,本人使用下载完后运行安装:直接默认安装即可。
2024-07-20 20:10:04
2021
原创 工业相机---初步认识+如何选型
线放大倍数M = 像元大小 / 系统分辨率 = 4.7um / 0.06mm = 0.078,焦距 = 150mm x 0.078 / (0.078 + 1)= 10.85,则选择焦距在10mm左右的镜头。物体光线经过镜头后的成像最大尺寸是可以确定的,如果成像最大尺寸小于感光靶面的尺寸,这样会导致感光靶面的外围会有一些像素点没有被照射到光线,从而会浪费感光靶面的像素点。,反差=(照度的最大值-照度的最小值)/(照度的最大值+照度的最小值),反差的数值总是
2024-07-05 17:26:19
1877
原创 pyinstaller打包onnxruntime-gpu报错找不到CUDA的解决方案
会在这个目录中查找自定义的钩子(hooks)文件。钩子文件用于在打包过程中处理一些特殊的依赖或行为。2、电脑上已经安装与onnxruntime-gpu对应版本的cuda和cudnn,并加入了环境变量。使用onnxruntime-gpu完成了深度学习模型部署,但在打包时发生了报错:找不到CUDA。--additional-hooks-dir=.:指定一个额外的目录,--onefile:将所有的文件和依赖打包成一个单独的可执行文件。-w main.py:指定需要打包的py文件,不带控制台窗口打包。
2024-07-04 10:15:32
2611
原创 NMS!!!
官方解释:NMS非极大值抑制(Non-Maximum Suppression, NMS),一个关键的后处理步骤,用于从检测器输出的多个边界框中筛选出最优的边界框。个人理解:模型推理输出后,会出现冗余框如下图所示。通过NMS操作可以去除冗余框,达到精准识别。
2024-06-30 18:16:04
600
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人