From 6b529d157c1540daca2478fddd71f4bd89fd9de9 Mon Sep 17 00:00:00 2001
From: manthandrive01 <78866140+manthandrive01@users.noreply.github.com>
Date: Fri, 12 Feb 2021 22:02:37 +0530
Subject: [PATCH] Colab demo is added

---
 demo/image_matting/Inference_with_ONNX/README.md         | 2 ++
 demo/image_matting/Inference_with_ONNX/inference_onnx.py | 8 ++++----
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/demo/image_matting/Inference_with_ONNX/README.md b/demo/image_matting/Inference_with_ONNX/README.md
index cb3f295..0403fb9 100644
--- a/demo/image_matting/Inference_with_ONNX/README.md
+++ b/demo/image_matting/Inference_with_ONNX/README.md
@@ -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
diff --git a/demo/image_matting/Inference_with_ONNX/inference_onnx.py b/demo/image_matting/Inference_with_ONNX/inference_onnx.py
index f19827e..cccfa23 100644
--- a/demo/image_matting/Inference_with_ONNX/inference_onnx.py
+++ b/demo/image_matting/Inference_with_ONNX/inference_onnx.py
@@ -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')
\ No newline at end of file
+    # 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')
\ No newline at end of file