Spaces:
Running
on
Zero
Running
on
Zero
Update inference_i2mv_sdxl.py
Browse files- 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)
|