Update MyPipe.py
Browse files
MyPipe.py
CHANGED
@@ -50,9 +50,9 @@ class RMBGPipe(Pipeline):
|
|
50 |
pil_im = Image.fromarray(result_image)
|
51 |
if return_mask ==True :
|
52 |
return pil_im
|
53 |
-
no_bg_image = Image.new("RGBA", pil_im.size, (0,0,0,0))
|
54 |
input_image = load_image(input_image)
|
55 |
-
no_bg_image.
|
|
|
56 |
return no_bg_image
|
57 |
|
58 |
# utilities functions
|
|
|
50 |
pil_im = Image.fromarray(result_image)
|
51 |
if return_mask ==True :
|
52 |
return pil_im
|
|
|
53 |
input_image = load_image(input_image)
|
54 |
+
no_bg_image = input_image.copy()
|
55 |
+
no_bg_image.putalpha(pil_im)
|
56 |
return no_bg_image
|
57 |
|
58 |
# utilities functions
|