mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-05 08:37:06 +08:00
718698a32a
* add yolov5cls * fixed bugs * fixed bugs * fixed preprocess bug * add yolov5cls readme * deal with comments * Add YOLOv5Cls Note * add yolov5cls test * add rvm support * support rvm model * add rvm demo * fixed bugs * add rvm readme * add TRT support * add trt support * add rvm test * add EXPORT.md * rename export.md * rm poros doxyen * deal with comments * deal with comments * add rvm video_mode note Co-authored-by: Jason <jiangjiajun@baidu.com> Co-authored-by: DefTruth <31974251+DefTruth@users.noreply.github.com>
MODNet准备部署模型
导出ONNX模型
访问MODNet官方github库，按照指引下载安装，下载模型文件，利用 onnx/export_onnx.py 得到onnx格式文件。
- 导出onnx格式文件
python -m onnx.export_onnx \ --ckpt-path=pretrained/modnet_photographic_portrait_matting.ckpt \ --output-path=pretrained/modnet_photographic_portrait_matting.onnx
下载预训练ONNX模型
为了方便开发者的测试，下面提供了MODNet导出的各系列模型，开发者可直接下载使用。（下表中模型的精度来源于源官方库）
| 模型 | 大小 | 精度 |
|---|---|---|
| modnet_photographic | 25MB | - |
| modnet_webcam | 25MB | - |
| modnet_photographic_256 | 25MB | - |
| modnet_webcam_256 | 25MB | - |
| modnet_photographic_512 | 25MB | - |
| modnet_webcam_512 | 25MB | - |
| modnet_photographic_1024 | 25MB | - |
| modnet_webcam_1024 | 25MB | - |
详细部署文档
版本说明
- 本版本文档和代码基于MODNet CommitID:28165a4 编写