kwaroran / abg-comfyui Goto Github PK

If someone wants to extract the background context for analysis, what can be done? This is a bit different from unCLIP and that text can be extracted rather than being a CLIP-based context https://github.com/pythongosssss/ComfyUI-WD14-Tagger

Side note: this also applies to RemBg for real life photos https://github.com/Jcd1230/rembg-comfyui-node

it works great if the batch size is one, but it will break if the batch size is more than one. This prevents it being used for cleaning up the final output of an animatediff flow. That would be AWESOME.

Here is the error message:

Error occurred when executing Remove Image Background (abg):

too many values to unpack (expected 2)

File "/Users/ross/apps/ComfyUI/execution.py", line 152, in recursive_execute
File "/Users/ross/apps/ComfyUI/execution.py", line 82, in get_output_data
uis = []

File "/Users/ross/apps/ComfyUI/execution.py", line 75, in map_node_over_list
nodes.before_node_execution()
^^^^^^^^^^^^^^^^^^^
File "/Users/ross/apps/ComfyUI/custom_nodes/abg-comfyui/init.py", line 57, in abg_remover
rmb = rmbg_fn(npa)
^^^^^^^^^^^^
File "/Users/ross/apps/ComfyUI/custom_nodes/abg-comfyui/init.py", line 31, in rmbg_fn
mask = get_mask(img)
^^^^^^^^^^^^^
File "/Users/ross/apps/ComfyUI/custom_nodes/abg-comfyui/init.py", line 9, in get_mask
h, w = h0, w0 = img.shape[:-1]
^^^^