mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-06 00:57:33 +08:00
cbf88a46fa
* 第一次提交 * 补充一处漏翻译 * deleted: docs/en/quantize.md * Update one translation * Update en version * Update one translation in code * Standardize one writing * Standardize one writing * Update some en version * Fix a grammer problem * Update en version for api/vision result * Merge branch 'develop' of https://github.com/charl-u/FastDeploy into develop * Checkout the link in README in vision_results/ to the en documents * Modify a title * Add link to serving/docs/ * Finish translation of demo.md * Update english version of serving/docs/ * Update title of readme * Update some links * Modify a title * Update some links * Update en version of java android README * Modify some titles * Modify some titles * Modify some titles * modify article to document * update some english version of documents in examples * Add english version of documents in examples/visions * Sync to current branch * Add english version of documents in examples * Add english version of documents in examples * Add english version of documents in examples * Update some documents in examples * Update some documents in examples * Update some documents in examples * Update some documents in examples * Update some documents in examples * Update some documents in examples * Update some documents in examples * Update some documents in examples * Update some documents in examples
55 lines
2.6 KiB
Markdown
55 lines
2.6 KiB
Markdown
English | [简体中文](README_CN.md)
|
||
# ResNet Ready-to-deploy Model
|
||
|
||
- ResNet Deployment is based on the code of [Torchvision](https://github.com/pytorch/vision/tree/v0.12.0) and [Pre-trained Models on ImageNet2012](https://github.com/pytorch/vision/tree/v0.12.0).
|
||
|
||
- （1）Deployment is conducted after [Export ONNX Model](#Export-the-ONNX-Model) by the *.pt provided by [Official Repository](https://github.com/pytorch/vision/tree/v0.12.0)；
|
||
- （2）The ResNet Model trained by personal data should [Export ONNX Model](#Export-the-ONNX-Model). Please refer to [Detailed Deployment Tutorials](#Detailed-Deployment-Documents) for deployment.
|
||
|
||
|
||
## Export the ONNX Model
|
||
|
||
|
||
Import [Torchvision](https://github.com/pytorch/vision/tree/v0.12.0), load the pre-trained model, and conduct model transformation as the following steps.
|
||
|
||
```python
|
||
import torch
|
||
import torchvision.models as models
|
||
|
||
model = models.resnet50(pretrained=True)
|
||
batch_size = 1 #Batch size
|
||
input_shape = (3, 224, 224) #Input data, and change to personal input shape
|
||
# #set the model to inference mode
|
||
model.eval()
|
||
x = torch.randn(batch_size, *input_shape) # Generate tensor
|
||
export_onnx_file = "ResNet50.onnx" # Purpose ONNX file name
|
||
torch.onnx.export(model,
|
||
x,
|
||
export_onnx_file,
|
||
opset_version=12,
|
||
input_names=["input"], # Input name
|
||
output_names=["output"], # Output name
|
||
dynamic_axes={"input":{0:"batch_size"}, # Batch variables
|
||
"output":{0:"batch_size"}})
|
||
```
|
||
|
||
## Download Pre-trained ONNX Model
|
||
|
||
For developers' testing, models exported by ResNet are provided below. Developers can download them directly. (The model accuracy in the following table is derived from the source official repository)
|
||
| Model | Size | Accuracy |
|
||
|:---------------------------------------------------------------- |:----- |:----- |
|
||
| [ResNet-18](https://bj.bcebos.com/paddlehub/fastdeploy/resnet18.onnx) | 45MB | |
|
||
| [ResNet-34](https://bj.bcebos.com/paddlehub/fastdeploy/resnet34.onnx) | 84MB | |
|
||
| [ResNet-50](https://bj.bcebos.com/paddlehub/fastdeploy/resnet50.onnx) | 98MB | |
|
||
| [ResNet-101](https://bj.bcebos.com/paddlehub/fastdeploy/resnet101.onnx) | 170MB | |
|
||
|
||
|
||
## Detailed Deployment Documents
|
||
|
||
- [Python Deployment](python)
|
||
- [C++ Deployment](cpp)
|
||
|
||
## Release Note
|
||
|
||
- Document and code are based on [Torchvision v0.12.0](https://github.com/pytorch/vision/tree/v0.12.0)
|