T2I-Dense-Face Dataset
收藏Dense-Face 数据集概述
数据集简介
Dense-Face 是一个用于个性化人脸生成的数据集,基于论文《Dense-Face: Personalized Face Generation Model via Dense Annotation Prediction》实现。该数据集包含来自 CASIA 和 CelebA 的人脸图像,并公开了 CASIA 部分的数据。
数据集内容
- T2I-Dense-Face 数据集:包含来自 CASIA 和 CelebA 的人脸图像。
- CASIA_tiny:可通过 Google Drive 下载,用于运行人脸生成模式。
- CASIA_small:包含 5%~10% 的完整数据集,可通过 Google Drive 下载。
- CASIA_full:完整数据集,需通过邮件联系 guoxia11@msu.edu 获取下载链接。
数据集使用
- 提供了
dataset_usage/*.ipynb文件,帮助用户理解和使用数据集。
预训练权重
- 提供了三种预训练权重,分别用于文本编辑模式、人脸生成模式和从头训练模式。可通过 Google Drive 下载。
数据集生成流程
- 文本编辑模式:使用
inference_code/stage_1_text_editing/stage_1_text_editing.ipynb进行文本编辑,生成结果保存在inference_code/output_stage_1。 - 生成条件:使用 MediaPipe 获取人脸区域和头部姿态条件,参考
./annotation_toolbox/dense_annotation_demo.ipynb。 - 人脸生成模式:基于步骤 2 生成的条件,修改步骤 1 的结果,生成最终的人脸图像,结果保存在
inference_code/output。
代码结构
bash ./Dense-Face ├── Dense-Face.yaml ├── inference_code │ ├── stage_1_text_editing/stage_1_text_editing.ipynb │ ├── inference.py │ ├── inference.sh │ ├── main.py │ ├── main.sh │ ├── reference_id │ ├── ropped_face │ ├── mask │ ├── output │ └── ... ├── annotation_toolbox │ ├── dense_annotation_demo.ipynb │ ├── Dense-Face-mediapipe.yml │ └── ... ├── dataset_usage │ └── readCelebAFacesDataset.ipynb └── test_samples
引用
如果使用该数据集,请引用以下论文: Bibtex @article{denseface, title={Dense-Face: Personalized Face Generation Model via Dense Annotation Prediction}, author={Xiao Guo and Manh Tran and Jiaxin Cheng and Xiaoming Liu}, journal={arXiv preprint arXiv:2412.18149}, year={2024} }




