This is my quick implementation of the SAL-VTON node for ComfyUI.
Based on the paper Keyu Y. Tingwei G. et al. (2023). Linking Garment with Person via Semantically Associated Landmakrs for Virtual Try-On
- Clone this repository into the
custom_nodes
folder. - Navigate to the cloned folder and run
pip install -r requirements.txt
. Be sure you're in the ComfyUI venv! - Download these landmark, warp, and salvton models.
- Create a folder, named
salvton
, in the ComfyUImodels
directory and copy all three downloaded models into it. - Profit?
- The garment should be 768x1024. If you don't have an image of the exact size, just resize it in ComfyUI.
- The results are poor if the background of the
person
image is not white. Consider using rembg or SAM to mask it and replace it with a white background. - The garment mask is just the shape of the input garment. You can generate it with SAM or use rembg like I did in the workflow.
This is a simple wrapper around the inference code available on ModelScope. Thanks to the original authors for their work.