OriLib commited on
Commit
bfa28aa
·
verified ·
1 Parent(s): 4f7b85b

Update MyPipe.py

Browse files
Files changed (1) hide show
  1. MyPipe.py +2 -2
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.paste(input_image, mask=pil_im)
 
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