Yet another custom node to detect human parts

Detect human parts using the DeepLabV3+ ResNet50 model from Keras-io. You can extract hair, arms, legs, and other parts with ease and with small memory usage.

This node aims to detect human parts using the model created by Keras-io. Their "Space" was impressive, and I wanted to use the model.

Unfortunately, the model uses an old Keras version, and there were no PyTorch implementation.

So I decided to convert the model to ONNX format and to create my HugginFace repository to share the model with the community.

Fortunately, Keras provides the model with a CC1.0 license, thank you guys to allow us to use it without any restriction.

Example

You can drag and drop the following image to try:

DeepLabV3+ ResNet50 for Human

Actually, all the model I found was not trained to detect human parts, but to detect some objects or urban elements. The Keras model is the only one I found that works!

Installation

I strongly recommend to use ComfyUI-Manager to install the node. It will install the dependencies and the model.

Note, as far as my repository isn't validated in the ComfyUI-Manager index, you must do the installation manually.

If you set up ComfyUI-Manager to "middle" or "weak" security, you can use the "Install from Git URL" feature.

# ensure that you have activated the virtual environment before !!

# then...
cd /path/to/your/ComfyUI/custom_nodes
git clone https://github.com/metal3d/ComfyUI_Human_Parts.git
cd ComfyUI_Human_Parts
pip install -r requirements.txt
# or
python -m pip install -r requirements.txt

# install the model
python install.py

Then, restart ComfyUI, refresh the UI, and you may find the "Human Parts mask generator" node.