pytorch模型转ncnn

最新推荐文章于 2024-08-13 08:16:12 发布

东城青年

最新推荐文章于 2024-08-13 08:16:12 发布

阅读量6.2k

点赞数 3

分类专栏： pytorch

本文链接：https://blog.csdn.net/qq_24946843/article/details/99629417

版权

pytorch 专栏收录该内容

31 篇文章 453 订阅 ¥9.90 ¥99.00

订阅专栏

本文介绍了如何将PyTorch模型转换为NCNN。首先，编译ncnn开源库，然后使用提供的pytorch2onnx脚本将PyTorch模型转换为ONNX。ONNX模型需要去除冗余维度以适配NCNN，最后通过onnx2ncnn工具生成ncnn所需的parm和bin文件。

摘要由CSDN通过智能技术生成

首先编译腾讯开源的ncnn：https://github.com/tongxiaobin/ncnn

在tools/pytorch里最下面有个链接：https://github.com/Tencent/ncnn/wiki/practical-pytorch-to-onnx-to-ncnn（将pytorch转为ncnn）

1、pytorch to onnx

新建pytorch2onnx.py文件,里面内容如下：

import torch
import torchvision
import torch.onnx

# An instance of your model
model = torchvision.models.resnet18()

# An example input you would normally provide to your model's forward() method
x = torch.rand(1, 3, 224, 224)

# Export the model
torch_out = torch.onnx._export(model, x, "res

了解本专栏