itsyogesh commited on
Commit
8cd1bb6
·
verified ·
1 Parent(s): 7c3ff1d

Remove smooth and denoising

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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(processed_mask).convert('L')
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))