Pytorch预训练模型文件转化为onnx格式

最新推荐文章于 2024-04-23 09:53:47 发布
最新推荐文章于 2024-04-23 09:53:47 发布
阅读量2k 收藏 3
点赞数 3
分类专栏： 各种问题 文章标签： PyTorch ONNX 模型转换 WACNet 模型部署
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_46773169/article/details/108889559
版权

代码如下

#coding: utf-8
import torch
from model.wacnet import WACNet # 导入自己构建好的模型
# from collections import OrderedDict

# 1.指定输入大小的shape
dummy_input = torch.randn(1, 3, 96, 96)

# 2. 导入模型
model = WACNet()

# 3. 载入模型参数, 'experiment/model_best.pt'为模型路径
# state_dict = torch.load('experiment/model_best.pt', map_location=lambda storage, loc: storage)
# new_state_dict = OrderedDict()
# for k, v in state_dict.items():
#     name = k[7:]  # remove 'module.' of DataParallel
#     new_state_dict[name] = v
model.load_state_dict(torch.load('experiment/model_best.pt', map_location='cpu'))

# 4.导出onnx模型文件
torch.onnx.export(model, dummy_input, "model_best.onnx",verbose=True)
ytao_wang
关注
专栏目录
用Hugging Face Transformers，高效部署：多显卡量化感知训练并转换为ONNX格式的多标签分类模型
q742971636的博客
06-26 495
import os])num_labels = 5 # 根据你的标签数量修改# 设置量化配置。
pytorch模型转onnx-量化rknn（bisenet）
橙子Blue不加冰
01-26 9282
1.pytorch模型转化onnx 先把pytorch的.pth模型转成onnx,例如我这个是用Bisenet转的 import argparse import os.path as osp import sys sys.path.insert(0, '.') import torch from lib.models import model_factory from configs import set_cfg_from_file torch.set_grad_enabled(False) pa
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）；
AdaBins pt模型转换为onnx
ApathyT的博客
10-18 655
def transform_pt_to_onnx(): MIN_DEPTH = 1e-3 MAX_DEPTH_NYU = 10 N_BINS = 256 batch_size = 1 input_shape = (3, 480, 640) pretrained_path = "./model/AdaBins_nyu.pt" onnx_output_path = "./model/AdaBins_nyu.onnx" device = t.
pt转onnx
Shaoyc
03-30 5736
pt模型转onnx模型完整流程
pytorch训练的pt模型转换为onnx（nn.DataParallel()、model、model.state_dict()）
weixin_44789335的博客
08-29 1983
1、读取pt模型文件，文件既可以是torch.save(model,path)整体保存的模型，也可以是保存的字典文件。2、torch.onnx.export()函数转换。
PyTorch 预训练模型转 onnx，Tensorrt 部署
qq_38593182的博客
07-28 1042
PyTorch 预训练模型转 onnx，Tensorrt 部署 所用环境 Centos 7 PyTorch 1.8.0（此为模型训练时的配置，转 onnx 时改为1.2.0） onnx 1.7.0 tensorrt 7.2.2.3 cuda 11.1 安装 onnx pip install onnx 安装 tensorrt 另一位博主 @幸福回头 的博文中 (https://blog.csdn.net/zt1091574181/article/details/100972562) 对于安装 tenso
pytorch权重文件pth转pt或onnx
random_repick的专栏
08-16 1755
pth pt onnx
将 tensorflow 版本的预训练 bert model 转化为 pytorch 版本.zip
01-07
在某些情况下，可能需要将一个框架中的预训练模型转换到另一个框架中，以利用其特定的优点。本文将详细介绍如何将 TensorFlow 版本的预训练 BERT 模型转化为 PyTorch 版本。 首先，BERT（Bidirectional Encoder ...
onnx模型部署 python_ONNX 介绍及TensorRT部署
weixin_39768247的博客
12-15 3015
ONNX简介Open Neural Network Exchange(ONNX，开放神经网络交换)格式，是一个用于表示深度学习模型的标准，可使模型在不同框架之间进行转移。【ONNX是一种针对机器学习所设计的开放式的文件格式，用于存储训练好的模型。它使得不同的人工智能框架(如Pytorch, MXNet)可以采用相同格式存储模型数据并交互。 ONNX的规范及代码主要由微软，亚马逊 ，Facebook...
Pytorch模型转onnx模型实例
01-20
如下所示： import io import torch import torch.onnx from models.C3AEModel import PlainC3AENetCBAM device = torch.device(cuda:0 if torch.cuda.is_available() else cpu) def test(): model = PlainC3AENetCBAM() pthfile = r'/home/joy/Projects/models/emotion/PlainC3AENet.pth' loaded_model = torch.load
pytorch模型(.pt)转onnx模型(.onnx)的方法详解(1)
热门推荐
Vertira的博客
10-30 3万+
pytorch模型(.pt)转onnx模型(.onnx)的方法详解(1)
可视化工具 netron & pt 转 onnx 格式
weixin_41258131的博客
08-25 4467
Netron 是一个开源的网络可视化工具，可以帮助开发人员和数据科学家可视化、理解和调试深度学习模型。
2、TensorRT学习笔记之PT转ONNX、可视化ONNX
最新发布
NOZhengYuan的博客
04-23 1647
主要讲解Pt权重转ONNX过程代码，以及如何可视化ONNX查看模型信息。
pytorch模型转化为onnx
xiaohuaibao的博客
03-18 3882
import torch, onnx, collections print('notice !!!! ----> use python3 run this script!!! \n') INPUT_DICT = 'res/model_test/eye_fbresnet10_motion.pth'//input path OUT_ONNX = 'res/model_test/eye_fbre...
pytorch模型转ONNX、并进行比较推理
qq128252的博客
03-27 2667
ONNX（Open Neural Network Exchange）是一种开放式的深度学习模型交换格式，旨在促进不同深度学习框架之间的互操作性。通过将深度学习模型转换为ONNX格式，可以将其从一个深度学习框架移植到另一个框架中，而无需重新训练模型或手动重新实现模型结构。ONNX是由微软和Facebook于2017年联合发布的，目前得到了众多公司和社区的支持和贡献。ONNX格式支持多种深度学习模型类型，包括卷积神经网络（CNN）、循环神经网络（RNN）和变换器（Transformer）等。
pytorch 转换onnx_有没有办法把经过预训练的模型从PyTorch转换成ONNX？
weixin_42559413的博客
01-14 331
我在我的自定义数据集上训练了StarGAN模型。我需要将这个模型从.pth(Pytorch)转换为.pb，以便在androidstudio上使用。我搜索了很多，找到了一些转换的方法。然而，所有的解决方案都不适用于我的案例。在我试过一个只有一个的小网络nn.线性层。在这个网络上，解决方案非常有效！在我想，我的网络包括Conv2D层和MaxPooling2D层，所以转换处理不起作用。在首先，这是我的网...
pytorch .bin模型文件怎么转化为onxx格式
06-01
你可以使用ONNX官方提供的工具将PyTorch模型文件转换为ONNX格式。具体步骤如下： 1. 首先安装ONNX包和PyTorch包： ``` pip install onnx pip install torch ``` 2. 加载PyTorch模型并将其转换为ONNX格式： ``` import torch import onnx # 加载PyTorch模型 model = torch.load('model.bin') # 将模型转换为ONNX格式 dummy_input = torch.randn(1, 3, 224, 224) input_names = ['input'] output_names = ['output'] onnx.export(model, dummy_input, 'model.onnx', input_names=input_names, output_names=output_names) ``` 在这个例子中，我们将PyTorch模型加载到变量`model`中，然后使用ONNX的`export`函数将模型转换为ONNX格式。我们还需要提供一个虚拟的输入张量`dummy_input`，以及输入和输出张量的名称。 3. 检查转换后的ONNX模型： ``` # 加载ONNX模型 onnx_model = onnx.load('model.onnx') # 检查模型是否正确 onnx.checker.check_model(onnx_model) ``` 使用`onnx.load`函数加载转换后的ONNX模型，并使用`onnx.checker.check_model`函数检查模型是否正确。如果检查成功，则表示模型已经成功转换为ONNX格式了。
评论 8
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值