pycharm里将.pt文件格式转化成onnx文件

已于 2024-09-13 16:47:24 修改
阅读量2.4k 收藏 11
点赞数 6
CC 4.0 BY-SA版权
文章标签： 人工智能 深度学习 yolov5
于 2024-09-13 13:37:32 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.youkuaiyun.com/qq_63600979/article/details/142207056

进行yolov5自定义模型训练后，形成.Pt文件，要想在rk3588开发板上部署yolov5的训练模型，需要将.pt文件先转化成onnx模型，最后再转成rknn完成部署

一、.pt文件转onnx文件

在配置好的环境中，在项目路径下输入: 

python export.py --weights runs/train/exp18/weights/best.pt --img-size 640 --batch-size 1（注意：一定是在你训练成功的环境，不放心就用prompt）

其中weights后面是你训练得到的best.pt文件的路径

转化完成后即可得到.onnx文件

yolov5s.pt 转 onnx 再转 ncnn 详细教程(一) | 粗浅解析 |【❤️yolov5s.pt 转 onnx 转 ncnn❤️】
专注于AI领域前沿技术学习与分享：目标检测、图像修复、超分重建、AI工程化
08-24 2811
文章目录v2.0/yolov5s.pt 转 onnx 再转 ncnn 具体步骤所运行代码版本yolov5 pt 模型转 onnxonnxsim 对其进行 simplifier 处理yolov5s_sim.onnx 转 ncnn对 ncnn.param 进行修正对 ncnn.param 进行 ncnnoptimize 优化推理验证ncnn模型【 yolov5s-opt.param 和 yolov5s-opt.bin】修改 ncnn/examples/yolov5.cppncnn 重新编译 ，运行总结
OpenCv —— Win10使用OpenCv调用Yolov5的.pt导出的.onnx模型进行测试【CPU版Cv】(附源码、附CV编译视频)
02-27 1449
OpenCv —— Win10使用OpenCv调用Yolov5的.pt导出的.onnx模型进行测试
【报错】.h5、.onnx、.pt文件的转化
2402_83131238的博客
02-20 552
对于这个问题，又仔细想了下，tensorboard是依赖，也就是之前pip uninstall没删干净的原因，于是重开了个环境，然后又开始报keras与python不兼容的错了，但是在链接里是匹配的，算了。2. .h5的输出与.onnx的不同【两个都是0. 1.，但是1的位置不同，总类别数为58，一个在18，一个在38，考虑从零开始计数，二者相加为总类别数】.pt和.h5文件之间无法直接转化，通常用.onnx作为桥梁，所以尝试.h5先转成.onnx，再转成.pt。记录一下碰到的问题。
yoloV5-6.1训练的pt模型转换为onnx模型
最新发布
haiyangmeng的博客
05-23 550
本文详细介绍了从YOLOv5模型训练到在树莓派4上测试的完整流程。首先，通过Anaconda和PyCharm配置环境，并准备数据集进行YOLOv5模型训练。接着，使用export.py脚本将训练好的.pt模型转换为.onnx格式。然后，安装OpenVINO库及其开发工具，利用模型优化器将.onnx模型转换为适用于OpenVINO的.bin和.xml格式。最后，计划在树莓派4上进行模型测试。整个过程涵盖了环境搭建、模型训练、格式转换及部署测试，为YOLOv5模型的开发和应用提供了全面的指导。
yolov5的.pt模型转.onnx模型并运行
weixin_45235219的博客
05-29 2005
运行export把pt转成onnx之前先做一下修改：data是训练.pt文件时的yaml文件，weights是需要转成onnx模型的pt模型的路径，batch-size改成1，想用cpu运行device就改成cpu，想用gpu运行就改成0，include就写onnx。直接在detect.py运行onnx模型：weights中放入onnx模型路径，如果想检测图片就在source放入图片路径，要检测摄像头就设置为0，data是训练.pt文件时的yaml文件。
YOLO模型的.pt文件转换为ONNX格式
fcgmqty的专栏
11-01 3773
如果你的Linux系统支持NVIDIA GPU，可以使用 TensorRT 将 ONNX 模型转换为 TensorRT 引擎文件（通常为 .bin 或 .engine），以便在 GPU 上加速推理。在将YOLO的ONNX模型转换为二进制的.bin文件并部署到Linux中运行，可以借助OpenVINO或TensorRT等工具完成。转换后的.engine文件可以通过TensorRT的API在Linux环境中运行，也可以在C++或Python代码中加载TensorRT引擎文件来执行推理。
把pt格式装换成ONNX格式
m0_69192635的博客
12-13 2267
将YOLOv5模型从.pt（PyTorch格式）转换为ONNX格式是一个常见的需求，特别是当你希望在不同的环境中部署模型时。以下是将YOLOv5的.pt文件转换为ONNX格式的步骤首先，确保你已经安装了必要的依赖项。
pt模型转onnx模型，onnx模型转engine模型，pt模型转engine模型详细教程（TensorRT，jetpack）
Mintary的博客
08-02 3080
背景是需要在nvidia jetpack4.5.1的arm64设备上跑yolov8,用TensorRT加速，需要用*.engine格式的模型，但是手头上的是pt格式模型，众所周知小板子的内存都很小，连安装ultralytics依赖库的容量都没有，所以我想到在windows系统上先把模型转好，然后再直接放在板子上用。
2、TensorRT学习笔记之PT转ONNX、可视化ONNX
NOZhengYuan的博客
04-23 2332
主要讲解Pt权重转ONNX过程代码，以及如何可视化ONNX查看模型信息。
pytorch训练图像分类模型pth转ONNX并测试
10-29
1. 搭建自己的简单二分类网络，使用pytorch训练和测试； 2. 将pytorch训练的pth模型转换成ONNX，并编码测试； 3. 含训练和测试数据，含训练ok的pth模型和ONNX模型，含完整python和C++实现； 4. 使用方法：首先运行“TrainTestConvertOnnx.py”执行“训练数据读入、模型训练、模型测试、导出onnx”，再运行“TestOnnx.cpp”测试onnx（需要配置OpenCV）；
YOLO系列模型 pt文件转化为ONNX导出
c_h_q_的博客
06-06 2176
YOLO系列模型 pt文件转化为ONNX导出
yolov5训练以及模型onnx转换
10-13
目标检测工程备份
【yolov5 pt文件转onnx并推理】
wo8wbs08的博客
11-24 2520
将pytorch训练的pt转换为onnx并执行推理
pycharm如何生成onnx文件
04-28
3. 导入ONNX库并使用ONNX模块中的`torch_onnx.export()`函数将模型转换为ONNX格式。例如： ``` import torch import onnx # Load the PyTorch model model = torch.load('model.pt') # Convert the model...
pt文件转onnx，保持标签关系以及动态输入输出（yolo）
qq_63385279的博客
02-07 2346
pt文件转onnx，保持标签关系以及动态输入输出（yolo）
将pt文件转换成onnx文件
A1983Z的博客
11-04 3062
将pt文件转化为onnx文件
yolo的模型转换权重pt文件转onnx（1）
m0_60657960的博客
03-08 3213
环境配置Ubuntu20.04 系统yolov5 5.0 训练标注的数据集，生成best.pt模型权重labelImg软件：用来标注数据集。
可视化工具 netron & pt 转 onnx 格式
weixin_41258131的博客
08-25 8371
Netron 是一个开源的网络可视化工具，可以帮助开发人员和数据科学家可视化、理解和调试深度学习模型。
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值