ComfyUI Extension: Batch Rembg for ComfyUI

Authored by Mamaaaamooooo

Created

Updated

29 stars

Run ComfyUI workflows without the setup

No installs, no CUDA version roulette, no GPU sitting idle on your bill. Bring a workflow and run it in the browser.

Remove background of plural images.

Looking for a different extension?

Custom Nodes (1)

README

batchImg-rembg

Rembg(Remove background) of image sequence for ComfyUI </br></br>

<img src = 'https://github.com/Mamaaaamooooo/batchImg-rembg-ComfyUI-nodes/assets/135937372/d3e05963-b047-4900-aa58-10f1e1b0980c' width="400" height="400"></img> <img src = 'https://github.com/Mamaaaamooooo/batchImg-rembg-ComfyUI-nodes/assets/135937372/bef5f8b4-3976-4c59-848f-7e77df6bd5a3' width="400" height="400"></img>

Installation

  1. Clone to your custom_nodes folder in ComfyUI:
git clone https://github.com/Mamaaaamooooo/batchImg-rembg-ComfyUI-nodes.git
  1. Install rembg[gpu] (recommended) or rembg, depending on GPU support, to your ComfyUI virtual environment. E.g.:
pip install rembg[gpu]
pip install tqdm

batchImg-rembg workflows will often make use of these helpful node packs:


</br>

Workflow

for example,

<img src= 'https://github.com/Mamaaaamooooo/batchImg-rembg-ComfyUI-nodes/assets/135937372/a516f83b-f149-45be-9dba-31c35c719f3b'>
</br> </br> </br> <img src = 'https://github.com/Mamaaaamooooo/batchImg-rembg-ComfyUI-nodes/assets/135937372/17966afa-0b8a-4774-95d0-2c57b3846694'>

Optional Models(Choose according to your work, downloaded automatically)


All models are downloaded and saved in the user home folder in the .u2net directory.

The available models are:

<details open> <summary>Rembg Model Name </summary>

| Name | Description | Link | |--------|-------------------------------------------------------------------------------------|------------------------------------| | u2net(default) | A pre-trained model for general use cases. | download, source | | u2netp | A lightweight version of u2net model. | download, source | | u2net_human_seg | A pre-trained model for human segmentation. | download, source | | u2net_cloth_seg | A pre-trained model for Cloths Parsing from human portrait. Here clothes are parsed into 3 category: Upper body, Lower body and Full body. | download, source | | silueta | Same as u2net but the size is reduced to 43Mb. | download, source | | isnet-general-use | A new pre-trained model for general use cases. |download, source | | isnet-anime | A high-accuracy segmentation for anime character. | download, source | | sam(not recommended, not easy to use) | A pre-trained model for any use cases. | download encoder, download decoder, source |

</details> </br> </br>

Example

<table class="center"> <tr style="line-height: 0"> <td width=34% style="border: none; text-align: center">Original Image</td> <td width=33% style="border: none; text-align: center">LineArt before Rembg</td> <td width=33% style="border: none; text-align: center">LineArt after Rembg</td> </tr> <tr> <td width=34% style="border: none"><img src="https://github.com/Mamaaaamooooo/batchImg-rembg-ComfyUI-nodes/assets/135937372/ef065506-e844-4b92-a282-d20198267f8e" style="width:100%"></td> <td width=33% style="border: none"><img src="https://github.com/Mamaaaamooooo/batchImg-rembg-ComfyUI-nodes/assets/135937372/e4440bec-d6dd-4726-8200-c38facbd1130" style="width:100%"></td> <td width=33% style="border: none"><img src="https://github.com/Mamaaaamooooo/batchImg-rembg-ComfyUI-nodes/assets/135937372/0d828694-1440-464b-9c68-7f646b73886f" style="width:100%"></td> </tr> </table> </br>

Acknowledgements

Thanks to rembg-comfyui-node for insight.</br> Thanks to rembg-comfyui-node-better for modifying repo.

Run ComfyUI workflows without the setup

No installs, no CUDA version roulette, no GPU sitting idle on your bill. Bring a workflow and run it in the browser.

Learn more