mirror
/
MODNet
mirror of https://github.com/ZHKKKe/MODNet.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Colab demo is added

revert-63-master
manthandrive01 2021-02-12 22:02:37 +05:30 committed by Zhanghan Ke
parent b9e3dddc86
commit 86ebafa2c5
2 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
demo/image_matting/Inference_with_ONNX/README.md
View File

@ -1,5 +1,7 @@
# Inference with onnxruntime
Please try MODNet image matting onnx-inference demo with [Colab Notebook](https://colab.research.google.com/drive/1P3cWtg8fnmu9karZHYDAtmm1vj1rgA-f?usp=sharing)
Download [modnet.onnx](https://drive.google.com/file/d/1cgycTQlYXpTh26gB9FTnthE7AvruV8hd/view?usp=sharing)
### 1. Export onnx model

8
demo/image_matting/Inference_with_ONNX/inference_onnx.py
View File

@ -110,7 +110,7 @@ if __name__ == '__main__':
# Optional - save png image without background
##############################################
im_PIL = Image.open(args.image_path)
matte = Image.fromarray(matte)
im_PIL.putalpha(matte) # add alpha channel to keep transparency
im_PIL.save('without_background.png')
# im_PIL = Image.open(args.image_path)
# matte = Image.fromarray(matte)
# im_PIL.putalpha(matte) # add alpha channel to keep transparency
# im_PIL.save('without_background.png')