Files
FastDeploy/examples/vision/matting/rvm/README.md
WJJ1995 718698a32a [Model] add RobustVideoMatting model (#400)
* 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>
2022-10-26 14:30:04 +08:00

1.2 KiB
Executable File
Raw Blame History

RobustVideoMatting 模型部署

模型版本说明

支持模型列表

目前FastDeploy支持如下模型的部署

下载预训练模型

为了方便开发者的测试下面提供了RobustVideoMatting导出的各系列模型开发者可直接下载使用。

模型 参数大小 精度 备注
rvm_mobilenetv3_fp32.onnx 15MB -
rvm_resnet50_fp32.onnx 103MB -
rvm_mobilenetv3_trt.onnx 15MB -
rvm_resnet50_trt.onnx 103MB -

Note

  • 如果要使用 TensorRT 进行推理,需要下载后缀为 trt 的 onnx 模型文件

详细部署文档