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]
^^^^