1580158
/
depthmap2mask
mirror of https://github.com/Extraltodeus/depthmap2mask.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #47 from 474172261/patch-1 

Update depth2image_depthmask.py
pull/51/head
Extraltodeus 2023-04-10 15:41:31 +02:00 committed by GitHub
parent 6150159cd6 4e6b349d8d
commit 1007e0513d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
scripts/depth2image_depthmask.py
View File

@ -31,6 +31,21 @@ class Script(scripts.Script):
def ui(self, is_img2img):
if not is_img2img: return
models = ["dpt_beit_large_512",
"dpt_beit_large_384",
"dpt_beit_base_384",
"dpt_swin2_large_384",
"dpt_swin2_base_384",
"dpt_swin2_tiny_256",
"dpt_swin_large_384",
"dpt_next_vit_large_384",
"dpt_levit_224",
"dpt_large_384",
"dpt_hybrid_384",
"midas_v21_384",
"midas_v21_small_256",
# "openvino_midas_v21_small_256"
]
treshold = gr.Slider(minimum=0, maximum=255, step=1, label='Contrasts cut level', value=0)
match_size = gr.Checkbox(label="Match input size",value=True)
@ -43,7 +58,7 @@ class Script(scripts.Script):
override_mask_blur = gr.Checkbox(label='Override mask blur to 0', value=True)
override_fill = gr.Checkbox(label='Override inpaint to original', value=True)
clean_cut = gr.Checkbox(label='Turn the depthmap into absolute black/white', value=False)
model_type = gr.Dropdown(label="Model", choices=['dpt_large','midas_v21','midas_v21_small'], value='midas_v21_small', type="index", elem_id="model_type")
model_type = gr.Dropdown(label="Model", choices=models, value="dpt_swin2_base_384", type="index", elem_id="model_type")
# model_type = gr.Dropdown(label="Model", choices=['dpt_large','dpt_hybrid','midas_v21','midas_v21_small'], value='dpt_large', type="index", elem_id="model_type")
return [save_depthmap,treshold,match_size,net_width,net_height,invert_depth,model_type,override_mask_blur,override_fill,clean_cut, save_alpha_crop]