From 78628e2f079bbb31410b8f186da50ba7f9bfaae6 Mon Sep 17 00:00:00 2001 From: channingss Date: Thu, 29 Aug 2019 17:18:22 +0800 Subject: [PATCH] update x2paddle_model_zoo.md for onnx model --- x2paddle_model_zoo.md | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/x2paddle_model_zoo.md b/x2paddle_model_zoo.md index bb0da1d..6b9b3ee 100644 --- a/x2paddle_model_zoo.md +++ b/x2paddle_model_zoo.md @@ -42,6 +42,7 @@ | Alexnet | [torchvison.model.alexnet](https://github.com/pytorch/vision/blob/master/torchvision/models/alexnet.py) |9| | Shufflenet | [onnx official](https://github.com/onnx/models/tree/master/vision/classification/shufflenet) |9| | Inception_v2 | [onnx official](https://github.com/onnx/models/tree/master/vision/classification/inception_and_googlenet/inception_v2) |9| +| Mobilenet_v2 | [pytorch(personal practice)](https://github.com/tonylins/pytorch-mobilenet-v2) |9| | Mnasnet_a1 | [pytorch(personal practice)](https://github.com/rwightman/gen-efficientnet-pytorch) |9| | Efficientnet_b0 | [pytorch(personal practice)](https://github.com/rwightman/gen-efficientnet-pytorch) |9| | squeezenet | [onnx official](https://s3.amazonaws.com/download.onnx/models/opset_9/squeezenet.tar.gz) |9| @@ -58,7 +59,7 @@ dummy_input = torch.randn(1, 3, 224, 224) #预训练后的pytorch model resnet18 = torchvision.models.resnet18(pretrained=True) -#"resnet18.onnx"为onnx model的存储路径 +#"resnet18.onnx"为onnx model的存储路径，1.1 torch.onnx.export(resnet18, dummy_input, "resnet18.onnx",verbose=True) ``` -- GitLab