English | 简体中文

MODNet准备部署模型

• MODNet
  • （1）官方库中提供的*.pt通过导出ONNX模型操作后，可进行部署；
  • （2）开发者基于自己数据训练的MODNet模型，可按照导出ONNX模型后，完成部署。

导出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导出的各系列模型，开发者可直接下载使用。（下表中模型的精度来源于源官方库）

详细部署文档

• Python部署
• C++部署

版本说明

• 本版本文档和代码基于MODNet CommitID:28165a4 编写