Pytorch模型转为onnx,onnx 模型 inference测试
1、mobilenet_v2生成.pt用torchvision导出mobilenet_v2网络结构,生成mobilenet_v2.ptimport torchfrom torch import nnfrom torchvision import modelsimport torch.nn.functional as Fclass MobileNet_v2(nn.Module): def __init__(self): super(MobileNet_v2, sel
原创
2021-12-25 20:08:16 ·
1514 阅读 ·
0 评论