lappemic commited on
Commit
c038f22
·
1 Parent(s): c2122de

fix median blur

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -16,7 +16,7 @@ def filters(file, box_blur, blur_pool2d, gaussian_blur2d, max_blur_pool2d, media
16
  x_out = K.filters.blur_pool2d(x_out, float(blur_pool2d))
17
  x_out = K.filters.gaussian_blur2d(x_out, float(gaussian_blur2d))
18
  x_out = K.filters.max_blur_pool2d(x_out, float(max_blur_pool2d))
19
- x_out = K.filters.median_blur(x_out, float(median_blur))
20
 
21
  return K.utils.tensor_to_image(x_out)
22
 
@@ -34,9 +34,9 @@ iface = gr.Interface(
34
  filters,
35
  [
36
  gr.inputs.Image(type="file"),
37
- gr.inputs.Slider(minimum=0, maximum=1, step=0.1, default=0, label="Box Blur"),
38
- gr.inputs.Slider(minimum=0, maximum=4, step=0.1, default=1, label="Blur Pool"),
39
- gr.inputs.Slider(minimum=0, maximum=4, step=0.1, default=1, label="Gaussian Blur"),
40
  gr.inputs.Slider(minimum=0, maximum=1, step=0.1, default=1, label="Max Pool"),
41
  gr.inputs.Slider(minimum=0, maximum=4, step=0.1, default=0, label="Median Blur"),
42
  ],
 
16
  x_out = K.filters.blur_pool2d(x_out, float(blur_pool2d))
17
  x_out = K.filters.gaussian_blur2d(x_out, float(gaussian_blur2d))
18
  x_out = K.filters.max_blur_pool2d(x_out, float(max_blur_pool2d))
19
+ x_out = K.filters.median_blur(x_out, (float(median_blur), float(median_blur)))
20
 
21
  return K.utils.tensor_to_image(x_out)
22
 
 
34
  filters,
35
  [
36
  gr.inputs.Image(type="file"),
37
+ gr.inputs.Slider(minimum=0, maximum=10, step=1, default=0, label="Box Blur"),
38
+ gr.inputs.Slider(minimum=0, maximum=10, step=1, default=1, label="Blur Pool"),
39
+ gr.inputs.Slider(minimum=0, maximum=20, step=1, default=1, label="Gaussian Blur"),
40
  gr.inputs.Slider(minimum=0, maximum=1, step=0.1, default=1, label="Max Pool"),
41
  gr.inputs.Slider(minimum=0, maximum=4, step=0.1, default=0, label="Median Blur"),
42
  ],