pytorch模型转onnx转ncnn
一、pth转onnx
训练生成好.pth文件后,参照网上的代码转onnx,网上代码如下
import torch
import torchvision
#define resnet18 model
model = torchvision.models.resnet18(pretrained=True)
#define input shape
x = torch.rand(1, 3, 224, 224)
#define input and output nodes, can be customized
in
原创
2020-05-18 14:38:05 ·
5284 阅读 ·
1 评论