ComfyUI_CSD_MT

ComfyUI_CSD_MT

CSD_MT is a method about 'Content-Style Decoupling for Unsupervised Makeup Transfer without Generating Pseudo Ground Truth', you can use it in comfyUI.

Tips

• This method need little space and VRAM(or CPU),all the models is about 200M(4 weights),and best quality size is 256*256
• 这个方法所用4个模型加起来不到200M,所需显存或CPU可以忽略不记,当然,图片输出最佳质量是256*256.

1. Installation

In the ./ComfyUI/custom_node directory, run the following:

git clone https://github.com/smthemex/ComfyUI_CSD_MT.git

---

2. Requirements

pip install -r requirements.txt

• When install dlib get error,use wheel to install it,download wheel from this address
• 如果按照dlib库失败，从此链接下载对应你python版本的轮子，用以下命令安装：

pip install dlib-19.24.1-cp311-cp311-win_amd64.whl

3.Model

• 3.1 download 3 checkpoints (79999_iter.pth,CSD_MT.pth,lms.dat) from google or baidu password:1d3e.从百度云或者Google下载三个模型
• 3.2 download 'resnet18-5c106cde.pth' from here 从链接下载resnet18-5c106cde.pth模型，不要改名字.

--  ComfyUI/models/CSDMT
    |-- 79999_iter.pth
    |-- CSD_MT.pth
    |--resnet18-5c106cde.pth #没有也会自动下载
--  ComfyUI/custom_node/ComfyUI_CSD_MT/quick_start/faceutils/dlibutils
    |--lms.dat  

4.Example

5.Citation

@inproceedings{sun2024content,
  title={Content-Style Decoupling for Unsupervised Makeup Transfer without Generating Pseudo Ground Truth},
  author={Sun, Zhaoyang and Xiong, Shengwu and Chen, Yaxiong and Rong, Yi}
  journal={Proceedings of the IEEE/CVF conference on computer vision and pattern recognition},
  year={2024}
}