mirror
/
MODNet
mirror of https://github.com/ZHKKKe/MODNet.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
MODNet/demo/image_matting/Inference_with_ONNX
History
manthandrive01 b9e3dddc86 Update README.md 2021-02-18 13:34:27 +08:00
..
README.md Update README.md 2021-02-18 13:34:27 +08:00
export_modnet_onnx.py Inference with onnx 2021-02-18 13:34:27 +08:00
inference_onnx.py Inference with onnx 2021-02-18 13:34:27 +08:00
requirements.txt Inference with onnx 2021-02-18 13:34:27 +08:00

README.md

Inference with onnxruntime

Download modnet.onnx

1. Export onnx model

Run the following command:

python export_modnet_onnx.py \
    --ckpt-path=pretrained/modnet_photographic_portrait_matting.ckpt \
    --output-path=modnet.onnx

2. Inference

Run the following command:

python inference_onnx.py \
    --image-path=PATH_TO_IMAGE \
    --output-path=matte.png \
    --model-path=modnet.onnx