Remove smooth and denoising
Browse files
app.py
CHANGED
|
@@ -174,10 +174,10 @@ def inference(image):
|
|
| 174 |
mask = predict(net, image_tensor, orig_size, hypar, device)
|
| 175 |
|
| 176 |
# Process the mask without cropping
|
| 177 |
-
processed_mask = smooth_and_denoise(mask)
|
| 178 |
|
| 179 |
# Convert processed mask to PIL image
|
| 180 |
-
pil_mask = Image.fromarray(
|
| 181 |
im_rgb = Image.open(image).convert("RGB")
|
| 182 |
|
| 183 |
im_dark = Image.new('RGB', im_rgb.size, (0, 0, 0))
|
|
|
|
| 174 |
mask = predict(net, image_tensor, orig_size, hypar, device)
|
| 175 |
|
| 176 |
# Process the mask without cropping
|
| 177 |
+
# processed_mask = smooth_and_denoise(mask)
|
| 178 |
|
| 179 |
# Convert processed mask to PIL image
|
| 180 |
+
pil_mask = Image.fromarray(mask).convert('L')
|
| 181 |
im_rgb = Image.open(image).convert("RGB")
|
| 182 |
|
| 183 |
im_dark = Image.new('RGB', im_rgb.size, (0, 0, 0))
|