Advertisement

Convert PyTorch models to NCNN and simultaneously support Android and iOS.

阅读量:

模型转换

pytorch转onnx,并用onnxruntime验证模型转换是否正确

复制代码
    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)
    checkpoint = torch.load(model_file)
    model.l

全部评论 (0)

还没有任何评论哟~