Ashoka74 commited on
Commit
0b62144
1 Parent(s): 7b38cfe

Update inference_i2mv_sdxl.py

Browse files
Files changed (1) hide show
  1. inference_i2mv_sdxl.py +2 -1
inference_i2mv_sdxl.py CHANGED
@@ -113,7 +113,8 @@ def remove_bg(image: Image.Image, net, transform, device, mask: Image.Image = No
113
 
114
  def preprocess_image(image: Image.Image, height, width):
115
  image = np.array(image)
116
- alpha = image[..., 3] > 0
 
117
  H, W = alpha.shape
118
  # get the bounding box of alpha
119
  y, x = np.where(alpha)
 
113
 
114
  def preprocess_image(image: Image.Image, height, width):
115
  image = np.array(image)
116
+ # alpha = image[..., 3] > 0
117
+ alpha = image
118
  H, W = alpha.shape
119
  # get the bounding box of alpha
120
  y, x = np.where(alpha)