├── LICENSE ├── README.md ├── app.py ├── attention.py ├── data ├── FGID.sh ├── FGID_caption.py ├── FGID_faceid_embeds.py ├── FGID_fuse_JSON.py └── FGID_mask.py ├── demo ├── controlnet_demo.py └── inpaint_demo.py ├── evaluation ├── EvaluationIMGs_stars_prompts.csv ├── convert_weights.py ├── eval_function.py └── style_template.py ├── examples ├── albert_einstein.jpg └── scarlett_johansson.jpg ├── functions.py ├── infer.py ├── infer_SDXL.py ├── models ├── BiSeNet │ ├── 6.jpg │ ├── __init__.py │ ├── evaluate.py │ ├── face_dataset.py │ ├── hair.png │ ├── logger.py │ ├── loss.py │ ├── makeup.py │ ├── makeup │ │ ├── 116_1.png │ │ ├── 116_3.png │ │ ├── 116_lip_ori.png │ │ └── 116_ori.png │ ├── model.py │ ├── modules │ │ ├── __init__.py │ │ ├── bn.py │ │ ├── deeplab.py │ │ ├── dense.py │ │ ├── functions.py │ │ ├── misc.py │ │ ├── residual.py │ │ └── src │ │ │ ├── checks.h │ │ │ ├── inplace_abn.cpp │ │ │ ├── inplace_abn.h │ │ │ ├── inplace_abn_cpu.cpp │ │ │ ├── inplace_abn_cuda.cu │ │ │ ├── inplace_abn_cuda_half.cu │ │ │ └── utils │ │ │ ├── checks.h │ │ │ ├── common.h │ │ │ └── cuda.cuh │ ├── optimizer.py │ ├── prepropess_data.py │ ├── resnet.py │ ├── test.py │ ├── train.py │ └── transform.py └── LLaVA │ └── demo_llva_use.py ├── pipelines ├── BaseConsistentID.py ├── StableDIffusionControlNetInpaint_ConsistentID.py └── StableDIffusionInpaint_ConsistentID.py ├── pipline_StableDiffusionXL_ConsistentID.py ├── pipline_StableDiffusion_ConsistentID.py ├── requirements.txt ├── train.py ├── train_SDXL.py ├── train_SDXL.sh ├── train_bash.sh ├── utils.py └── utils_SDXL.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/README.md -------------------------------------------------------------------------------- /app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/app.py -------------------------------------------------------------------------------- /attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/attention.py -------------------------------------------------------------------------------- /data/FGID.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/data/FGID.sh -------------------------------------------------------------------------------- /data/FGID_caption.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/data/FGID_caption.py -------------------------------------------------------------------------------- /data/FGID_faceid_embeds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/data/FGID_faceid_embeds.py -------------------------------------------------------------------------------- /data/FGID_fuse_JSON.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/data/FGID_fuse_JSON.py -------------------------------------------------------------------------------- /data/FGID_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/data/FGID_mask.py -------------------------------------------------------------------------------- /demo/controlnet_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/demo/controlnet_demo.py -------------------------------------------------------------------------------- /demo/inpaint_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/demo/inpaint_demo.py -------------------------------------------------------------------------------- /evaluation/EvaluationIMGs_stars_prompts.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/evaluation/EvaluationIMGs_stars_prompts.csv -------------------------------------------------------------------------------- /evaluation/convert_weights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/evaluation/convert_weights.py -------------------------------------------------------------------------------- /evaluation/eval_function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/evaluation/eval_function.py -------------------------------------------------------------------------------- /evaluation/style_template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/evaluation/style_template.py -------------------------------------------------------------------------------- /examples/albert_einstein.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/examples/albert_einstein.jpg -------------------------------------------------------------------------------- /examples/scarlett_johansson.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/examples/scarlett_johansson.jpg -------------------------------------------------------------------------------- /functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/functions.py -------------------------------------------------------------------------------- /infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/infer.py -------------------------------------------------------------------------------- /infer_SDXL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/infer_SDXL.py -------------------------------------------------------------------------------- /models/BiSeNet/6.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/models/BiSeNet/6.jpg -------------------------------------------------------------------------------- /models/BiSeNet/__init__.py: -------------------------------------------------------------------------------- 1 | #__init__.py 2 | # from BiSeNet.model import * -------------------------------------------------------------------------------- /models/BiSeNet/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/models/BiSeNet/evaluate.py -------------------------------------------------------------------------------- /models/BiSeNet/face_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/models/BiSeNet/face_dataset.py -------------------------------------------------------------------------------- /models/BiSeNet/hair.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/models/BiSeNet/hair.png -------------------------------------------------------------------------------- /models/BiSeNet/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/models/BiSeNet/logger.py -------------------------------------------------------------------------------- /models/BiSeNet/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/models/BiSeNet/loss.py -------------------------------------------------------------------------------- /models/BiSeNet/makeup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/models/BiSeNet/makeup.py -------------------------------------------------------------------------------- /models/BiSeNet/makeup/116_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/models/BiSeNet/makeup/116_1.png -------------------------------------------------------------------------------- /models/BiSeNet/makeup/116_3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/models/BiSeNet/makeup/116_3.png -------------------------------------------------------------------------------- /models/BiSeNet/makeup/116_lip_ori.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/models/BiSeNet/makeup/116_lip_ori.png -------------------------------------------------------------------------------- /models/BiSeNet/makeup/116_ori.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/models/BiSeNet/makeup/116_ori.png -------------------------------------------------------------------------------- /models/BiSeNet/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/models/BiSeNet/model.py -------------------------------------------------------------------------------- /models/BiSeNet/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/models/BiSeNet/modules/__init__.py -------------------------------------------------------------------------------- /models/BiSeNet/modules/bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/models/BiSeNet/modules/bn.py -------------------------------------------------------------------------------- /models/BiSeNet/modules/deeplab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/models/BiSeNet/modules/deeplab.py -------------------------------------------------------------------------------- /models/BiSeNet/modules/dense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/models/BiSeNet/modules/dense.py -------------------------------------------------------------------------------- /models/BiSeNet/modules/functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/models/BiSeNet/modules/functions.py -------------------------------------------------------------------------------- /models/BiSeNet/modules/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/models/BiSeNet/modules/misc.py -------------------------------------------------------------------------------- /models/BiSeNet/modules/residual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/models/BiSeNet/modules/residual.py -------------------------------------------------------------------------------- /models/BiSeNet/modules/src/checks.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/models/BiSeNet/modules/src/checks.h -------------------------------------------------------------------------------- /models/BiSeNet/modules/src/inplace_abn.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/models/BiSeNet/modules/src/inplace_abn.cpp -------------------------------------------------------------------------------- /models/BiSeNet/modules/src/inplace_abn.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/models/BiSeNet/modules/src/inplace_abn.h -------------------------------------------------------------------------------- /models/BiSeNet/modules/src/inplace_abn_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/models/BiSeNet/modules/src/inplace_abn_cpu.cpp -------------------------------------------------------------------------------- /models/BiSeNet/modules/src/inplace_abn_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/models/BiSeNet/modules/src/inplace_abn_cuda.cu -------------------------------------------------------------------------------- /models/BiSeNet/modules/src/inplace_abn_cuda_half.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/models/BiSeNet/modules/src/inplace_abn_cuda_half.cu -------------------------------------------------------------------------------- /models/BiSeNet/modules/src/utils/checks.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/models/BiSeNet/modules/src/utils/checks.h -------------------------------------------------------------------------------- /models/BiSeNet/modules/src/utils/common.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/models/BiSeNet/modules/src/utils/common.h -------------------------------------------------------------------------------- /models/BiSeNet/modules/src/utils/cuda.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/models/BiSeNet/modules/src/utils/cuda.cuh -------------------------------------------------------------------------------- /models/BiSeNet/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/models/BiSeNet/optimizer.py -------------------------------------------------------------------------------- /models/BiSeNet/prepropess_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/models/BiSeNet/prepropess_data.py -------------------------------------------------------------------------------- /models/BiSeNet/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/models/BiSeNet/resnet.py -------------------------------------------------------------------------------- /models/BiSeNet/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/models/BiSeNet/test.py -------------------------------------------------------------------------------- /models/BiSeNet/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/models/BiSeNet/train.py -------------------------------------------------------------------------------- /models/BiSeNet/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/models/BiSeNet/transform.py -------------------------------------------------------------------------------- /models/LLaVA/demo_llva_use.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/models/LLaVA/demo_llva_use.py -------------------------------------------------------------------------------- /pipelines/BaseConsistentID.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/pipelines/BaseConsistentID.py -------------------------------------------------------------------------------- /pipelines/StableDIffusionControlNetInpaint_ConsistentID.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/pipelines/StableDIffusionControlNetInpaint_ConsistentID.py -------------------------------------------------------------------------------- /pipelines/StableDIffusionInpaint_ConsistentID.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/pipelines/StableDIffusionInpaint_ConsistentID.py -------------------------------------------------------------------------------- /pipline_StableDiffusionXL_ConsistentID.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/pipline_StableDiffusionXL_ConsistentID.py -------------------------------------------------------------------------------- /pipline_StableDiffusion_ConsistentID.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/pipline_StableDiffusion_ConsistentID.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/requirements.txt -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/train.py -------------------------------------------------------------------------------- /train_SDXL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/train_SDXL.py -------------------------------------------------------------------------------- /train_SDXL.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/train_SDXL.sh -------------------------------------------------------------------------------- /train_bash.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/train_bash.sh -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/utils.py -------------------------------------------------------------------------------- /utils_SDXL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackAILab/ConsistentID/HEAD/utils_SDXL.py --------------------------------------------------------------------------------