modnet

English | 简体中文

MODNet Ready-to-deploy Model

• MODNet
  • （1）The *.pt provided by the Official Library can be deployed after Export ONNX Model；
  • （2）As for MODNet model trained on customized data, please follow the operations guidelines in Export ONNX Model to complete the deployment.

Export ONNX Model

Visit MODNet official github repository, follow the guidelines to download model files, and employ onnx/export_onnx.py to get files in onnx format.

• Export files in onnx format

  python -m onnx.export_onnx \
    --ckpt-path=pretrained/modnet_photographic_portrait_matting.ckpt \
    --output-path=pretrained/modnet_photographic_portrait_matting.onnx

Download Pre-trained ONNX Model

For developers' testing, models exported by MODNet are provided below. Developers can download them directly. (The accuracy in the following table is derived from the source official repository)

Model	Size	Accuracy
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	-

Detailed Deployment Tutorials

• Python Deployment
• C++ Deployment

Release Note

• This tutorial and related code are written based on MODNet CommitID:28165a4