add default alpha_matting
Browse files
app.py
CHANGED
|
@@ -28,6 +28,15 @@ def remove_background(input_image, bg_color, model_name, alpha_matting, post_pro
|
|
| 28 |
"post_process_mask": post_process_mask,
|
| 29 |
"only_mask": only_mask
|
| 30 |
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 31 |
|
| 32 |
# Use the remove function
|
| 33 |
if session or bg_color_rgba:
|
|
|
|
| 28 |
"post_process_mask": post_process_mask,
|
| 29 |
"only_mask": only_mask
|
| 30 |
}
|
| 31 |
+
|
| 32 |
+
# Add alpha matting parameters if enabled
|
| 33 |
+
if alpha_matting:
|
| 34 |
+
remove_kwargs.update({
|
| 35 |
+
"alpha_matting": True,
|
| 36 |
+
"alpha_matting_foreground_threshold": 270,
|
| 37 |
+
"alpha_matting_background_threshold": 20,
|
| 38 |
+
"alpha_matting_erode_size": 11
|
| 39 |
+
})
|
| 40 |
|
| 41 |
# Use the remove function
|
| 42 |
if session or bg_color_rgba:
|