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

pull/63/head
manthandrive01 2021-02-12 22:02:37 +05:30
parent 74a874e357
commit 6b529d157c
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')