diff --git a/README.md b/README.md index de0a469..a8f24f4 100644 --- a/README.md +++ b/README.md @@ -5,5 +5,50 @@ * 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. -# Example +# 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](https://github.com/z-mahmud22/Dlib_Windows_Python3.x) +* 如果按照dlib库失败,从此链接下载对应你python版本的轮子,用以下命令安装: + +``` +pip install dlib-19.24.1-cp311-cp311-win_amd64.whl +``` + +# 3.Model +* 3.1 download 2 checkpoints (79999_iter.pth,CSD_MT.pth) from [google](https://drive.google.com/drive/folders/1pvSgkpsb7k6Ph1_oCmFkMQZPgL7PaTO0) or [baidu](https://pan.baidu.com/share/init?surl=C7K4xk5W0X65yUQh41AmfQ) password:1d3e.从百度云或者Google下载三个模型 +* 3.2 download 'resnet18-5c106cde.pth' from [here](https://download.pytorch.org/models/resnet18-5c106cde.pth) 从链接下载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 # Already in the project, no need to download,已内置在插件中,不用下载 +``` + +# 4.Example ![](https://github.com/smthemex/ComfyUI_CSD_MT/blob/main/example.png) + + +# 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} +} +```