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
|