Spaces:
Running
Running
yizhangliu
commited on
Commit
•
4b39dfc
1
Parent(s):
46bf436
Update app.py
Browse files
app.py
CHANGED
@@ -186,7 +186,7 @@ def model_process(image, mask):
|
|
186 |
# logger.info(f"Resized image shape: {image.shape}")
|
187 |
print(f"Resized image shape: {image.shape} / {image[250][250]}")
|
188 |
|
189 |
-
mask, _ = load_img(mask
|
190 |
# mask = np.array(mask_pil)
|
191 |
mask = resize_max_size(mask, size_limit=size_limit, interpolation=interpolation)
|
192 |
print(f"mask image shape: {mask.shape} / {type(mask)} / {mask[250][250]}")
|
|
|
186 |
# logger.info(f"Resized image shape: {image.shape}")
|
187 |
print(f"Resized image shape: {image.shape} / {image[250][250]}")
|
188 |
|
189 |
+
mask, _ = load_img(mask, gray=True)
|
190 |
# mask = np.array(mask_pil)
|
191 |
mask = resize_max_size(mask, size_limit=size_limit, interpolation=interpolation)
|
192 |
print(f"mask image shape: {mask.shape} / {type(mask)} / {mask[250][250]}")
|