AI/rembg
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/danielgatis/rembg synced 2025-08-19 20:29:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add controls to Gradio interface (#581)

Browse Source 
Co-authored-by: Zebradots <projects@zebradots.com>
This commit is contained in:
zebradots 2024-01-31 07:05:05 -06:00 committed by GitHub
parent 49d1686f65
commit cf962f5429
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
rembg/commands/s_command.py
View File

@ -264,7 +264,15 @@ def s_command(port: int, host: str, log_level: str, threads: int) -> None:
def inference(input_path, model, cmd_args):
output_path = "output.png"
kwargs = {}
kwargs = {
"alpha_matting": a,
"alpha_matting_foreground_threshold": af,
"alpha_matting_background_threshold": ab,
"alpha_matting_erode_size": ae,
"only_mask": om,
"post_process_mask": ppm,
}
if cmd_args:
kwargs.update(json.loads(cmd_args))
kwargs["session"] = new_session(model, **kwargs)
@ -281,6 +289,12 @@ def s_command(port: int, host: str, log_level: str, threads: int) -> None:
[
gr.components.Image(type="filepath", label="Input"),
gr.components.Dropdown(sessions_names, value="u2net", label="Models"),
gr.components.Checkbox(value=True, label="Alpha matting"),
gr.components.Slider(value=240, minimum=0, maximum=255, label="Foreground threshold"),
gr.components.Slider(value=10, minimum=0, maximum=255, label="Background threshold"),
gr.components.Slider(value=40, minimum=0, maximum=255, label="Erosion size"),
gr.components.Checkbox(value=False, label="Only mask"),
gr.components.Checkbox(value=True, label="Post process mask"),
gr.components.Textbox(label="Arguments"),
],
gr.components.Image(type="filepath", label="Output"),