nodes/spinagon-ComfyUI-seam-carving
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Reduce step size (better respect mask)

Browse Source
This commit is contained in:
flak (home)
2023-10-13 10:20:02 +03:00
parent 51fe540e19
commit f81d3d4286
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
seamcarving.py
+2 -1
View File
@@ -60,7 +60,8 @@ class SeamCarving:
energy_mode="forward", # choose from {backward, forward} energy_mode="forward", # choose from {backward, forward}
order="width-first", # choose from {width-first, height-first} order="width-first", # choose from {width-first, height-first}
keep_mask=k_mask, # object mask to protect from removal keep_mask=k_mask, # object mask to protect from removal
drop_mask=d_mask drop_mask=d_mask,
step_ratio=0.1
) )
results.append(torch.from_numpy(dst)) results.append(torch.from_numpy(dst))
return (torch.stack(results),) return (torch.stack(results),)