Advertisement

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)

还没有任何评论哟~