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 bffa00eebd Inference with onnx 
export model and run inference
 		2021-02-11 23:52:28 +05:30
..
README.md Inference with onnx 2021-02-11 23:52:28 +05:30
export_modnet_onnx.py Inference with onnx 2021-02-11 23:52:28 +05:30
inference_onnx.py Inference with onnx 2021-02-11 23:52:28 +05:30
requirements.txt Inference with onnx 2021-02-11 23:52:28 +05:30

README.md

Inference with onnxruntime

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