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

Revert "Colab demo is added" 

This reverts commit 86ebafa2c5.
revert-63-master
Zhanghan Ke 2021-02-18 13:35:16 +08:00 committed by GitHub
parent 86ebafa2c5
commit 72c6e919a0
2 changed files with 4 additions and 6 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,7 +1,5 @@
# 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')