zhanshichao / TinyBeauty

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

TinyBeauty: Toward Tiny and High-quality Facial Makeup with Data Amplify Learning

Accepted by ECCV2024

Our TinyBeauty effectively synthesizes stunning makeup styles with consistent content, enabling seamless video application.

Release

Visual Results

High-quality Facial Makeup

Our DDA generates consistent makeup styles while retain the facial content and identity of the original image

Facial makeup results on high-resolution (1024*1024) images.

Comparison with Previous Works

Visual comparison of TineBeauty and competing methods on the FFHQ Dataset.

Visual comparison of TineBeauty and competing methods on the MT Dataset.

Visual comparison of TineBeauty and BeautyREC on challenging out-of-distribution examples

Usage

Prepare data

Download sample image pair and makeup style template from here, and place it in the ./data folder.

Finetune

python SD_finetune.py 
    -m runwayml/stable-diffusion-v1-5
    -e h94/IP-Adapter
    -nonmakeup data/Finetune_Data/train
    -makeup data/Finetune_Data/train_purple
    -o "$LORA_MODEL_SAVE_PATH"

Inference

python SD_inference.py
    -m "$LORA_MODEL_SAVE_PATH"
    -s data/Finetune_Data/purple.png
    -d data/Finetune_Data/test
    -o res/test1

Cite

If you find TinyBeauty useful for your research and applications, please cite us using this BibTeX:

@misc{jin2024tiny,
      title={Toward Tiny and High-quality Facial Makeup with Data Amplify Learning}, 
      author={Qiaoqiao Jin and Xuanhong Chen and Meiguang Jin and Ying Cheng and Rui Shi and Yucheng Zheng and Yupeng Zhu and Bingbing Ni},
      year={2024},
      eprint={2403.15033},
      archivePrefix={arXiv},
      primaryClass={cs.CV}
}

About


Languages

Language:Python 98.2%Language:JavaScript 1.5%Language:HTML 0.3%Language:CSS 0.0%