feat: add inpaint mask generation functionality

uses rembg, see https://github.com/danielgatis/rembg
This commit is contained in:
Manuel Schmid
2024-01-24 02:22:50 +01:00
parent f6d67d7655
commit db5512fb93
5 changed files with 44 additions and 3 deletions
+6
View File
@@ -318,6 +318,12 @@ example_inpaint_prompts = get_config_item_or_set_default(
example_inpaint_prompts = [[x] for x in example_inpaint_prompts]
default_inpaint_mask_model = get_config_item_or_set_default(
key='default_inpaint_mask_model',
default_value='isnet-general-use',
validator=lambda x: isinstance(x, str)
)
config_dict["default_loras"] = default_loras = default_loras[:5] + [['None', 1.0] for _ in range(5 - len(default_loras))]
possible_preset_keys = [
+4
View File
@@ -35,6 +35,10 @@ default_parameters = {
inpaint_engine_versions = ['None', 'v1', 'v2.5', 'v2.6']
performance_selections = ['Speed', 'Quality', 'Extreme Speed']
inpaint_mask_models = [
'u2net', 'u2netp', 'u2net_human_seg', 'u2net_cloth_seg', 'silueta', 'isnet-general-use', 'isnet-anime', 'sam'
]
inpaint_option_default = 'Inpaint or Outpaint (default)'
inpaint_option_detail = 'Improve Detail (face, hand, eyes, etc.)'
inpaint_option_modify = 'Modify Content (add objects, change background, etc.)'