pytorch模型转换为ncnn模型,并部署到android平台
发布时间
阅读量:
阅读量
模型转换
pytorch转onnx
import torch.utils.data
from torch.autograd import Variable
from squeezenet import squeezenet1_2
import onnxruntime
import numpy as np
from onnxruntime.datasets import get_example
import cv2
import onnx
model_file = '/home/bbt/qinghua/detetion/pytorch-mobilenet/model/model_best_squeezenet1_2_age0.926.pth.tar'
num_class =3
# create model
model=squeezenet1_2(pretrained=False,num_classes=num_class)
model = torch.nn.DataParallel(model)
# optionally resume from a checkpoint
checkpoint = torch.load(model_fi
全部评论 (0)
还没有任何评论哟~
