├── .gitignore ├── .vscode └── settings.json ├── README.md ├── cleaner.py ├── conf └── conf.yaml ├── dataset.py ├── dloss.py ├── document ├── apex.md ├── images │ ├── 1.jpg │ ├── 2.jpg │ └── 3.jpg └── install.md ├── facedancer.py ├── lib └── loss.py ├── loss.py ├── main.py ├── makedata.py ├── model.py ├── models ├── activation.py ├── bfm.py ├── ca.py ├── cbam.py ├── clip │ ├── __init__.py │ ├── bpe_simple_vocab_16e6.txt.gz │ ├── clip.py │ ├── model.py │ └── simple_tokenizer.py ├── constants.py ├── diff_augment.py ├── dino_discriminator.py ├── discriminator.py ├── face_models │ ├── arcface.py │ └── iresnet.py ├── faceparser.py ├── gen2.py ├── gen3.py ├── gen4.py ├── generator.py ├── gennew.py ├── gradnorm.py ├── load_mats.py ├── mobile_style_generator.py ├── modulated_conv2d.py ├── multiscalediscriminator.py ├── pretrained_builder.py ├── projector.py ├── reconnet.py ├── resnet.py ├── shared.py ├── stargandiscriminator.py ├── vit.py └── vit_utils.py ├── prepare_data.py ├── test.py ├── torch_utils ├── __init__.py ├── custom_ops.py ├── gen_utils.py ├── misc.py ├── ops │ ├── __init__.py │ ├── bias_act.cpp │ ├── bias_act.cu │ ├── bias_act.h │ ├── bias_act.py │ ├── conv2d_gradfix.py │ ├── conv2d_resample.py │ ├── filtered_lrelu.cpp │ ├── filtered_lrelu.cu │ ├── filtered_lrelu.h │ ├── filtered_lrelu.py │ ├── filtered_lrelu_ns.cu │ ├── filtered_lrelu_rd.cu │ ├── filtered_lrelu_wr.cu │ ├── fma.py │ ├── grid_sample_gradfix.py │ ├── upfirdn2d.cpp │ ├── upfirdn2d.cu │ ├── upfirdn2d.h │ └── upfirdn2d.py ├── persistence.py ├── training_stats.py └── utils_spectrum.py ├── trainer.py ├── ui ├── FaceTool.ui └── ui_FaceTool.py └── uic └── FaceTool.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/.gitignore -------------------------------------------------------------------------------- /.vscode/settings.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/.vscode/settings.json -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/README.md -------------------------------------------------------------------------------- /cleaner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/cleaner.py -------------------------------------------------------------------------------- /conf/conf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/conf/conf.yaml -------------------------------------------------------------------------------- /dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/dataset.py -------------------------------------------------------------------------------- /dloss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/dloss.py -------------------------------------------------------------------------------- /document/apex.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/document/apex.md -------------------------------------------------------------------------------- /document/images/1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/document/images/1.jpg -------------------------------------------------------------------------------- /document/images/2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/document/images/2.jpg -------------------------------------------------------------------------------- /document/images/3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/document/images/3.jpg -------------------------------------------------------------------------------- /document/install.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/document/install.md -------------------------------------------------------------------------------- /facedancer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/facedancer.py -------------------------------------------------------------------------------- /lib/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/lib/loss.py -------------------------------------------------------------------------------- /loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/loss.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/main.py -------------------------------------------------------------------------------- /makedata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/makedata.py -------------------------------------------------------------------------------- /model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/model.py -------------------------------------------------------------------------------- /models/activation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/models/activation.py -------------------------------------------------------------------------------- /models/bfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/models/bfm.py -------------------------------------------------------------------------------- /models/ca.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/models/ca.py -------------------------------------------------------------------------------- /models/cbam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/models/cbam.py -------------------------------------------------------------------------------- /models/clip/__init__.py: -------------------------------------------------------------------------------- 1 | from .clip import * 2 | -------------------------------------------------------------------------------- /models/clip/bpe_simple_vocab_16e6.txt.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/models/clip/bpe_simple_vocab_16e6.txt.gz -------------------------------------------------------------------------------- /models/clip/clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/models/clip/clip.py -------------------------------------------------------------------------------- /models/clip/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/models/clip/model.py -------------------------------------------------------------------------------- /models/clip/simple_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/models/clip/simple_tokenizer.py -------------------------------------------------------------------------------- /models/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/models/constants.py -------------------------------------------------------------------------------- /models/diff_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/models/diff_augment.py -------------------------------------------------------------------------------- /models/dino_discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/models/dino_discriminator.py -------------------------------------------------------------------------------- /models/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/models/discriminator.py -------------------------------------------------------------------------------- /models/face_models/arcface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/models/face_models/arcface.py -------------------------------------------------------------------------------- /models/face_models/iresnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/models/face_models/iresnet.py -------------------------------------------------------------------------------- /models/faceparser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/models/faceparser.py -------------------------------------------------------------------------------- /models/gen2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/models/gen2.py -------------------------------------------------------------------------------- /models/gen3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/models/gen3.py -------------------------------------------------------------------------------- /models/gen4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/models/gen4.py -------------------------------------------------------------------------------- /models/generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/models/generator.py -------------------------------------------------------------------------------- /models/gennew.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/models/gennew.py -------------------------------------------------------------------------------- /models/gradnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/models/gradnorm.py -------------------------------------------------------------------------------- /models/load_mats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/models/load_mats.py -------------------------------------------------------------------------------- /models/mobile_style_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/models/mobile_style_generator.py -------------------------------------------------------------------------------- /models/modulated_conv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/models/modulated_conv2d.py -------------------------------------------------------------------------------- /models/multiscalediscriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/models/multiscalediscriminator.py -------------------------------------------------------------------------------- /models/pretrained_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/models/pretrained_builder.py -------------------------------------------------------------------------------- /models/projector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/models/projector.py -------------------------------------------------------------------------------- /models/reconnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/models/reconnet.py -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/models/resnet.py -------------------------------------------------------------------------------- /models/shared.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/models/shared.py -------------------------------------------------------------------------------- /models/stargandiscriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/models/stargandiscriminator.py -------------------------------------------------------------------------------- /models/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/models/vit.py -------------------------------------------------------------------------------- /models/vit_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/models/vit_utils.py -------------------------------------------------------------------------------- /prepare_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/prepare_data.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/test.py -------------------------------------------------------------------------------- /torch_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/torch_utils/__init__.py -------------------------------------------------------------------------------- /torch_utils/custom_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/torch_utils/custom_ops.py -------------------------------------------------------------------------------- /torch_utils/gen_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/torch_utils/gen_utils.py -------------------------------------------------------------------------------- /torch_utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/torch_utils/misc.py -------------------------------------------------------------------------------- /torch_utils/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/torch_utils/ops/__init__.py -------------------------------------------------------------------------------- /torch_utils/ops/bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/torch_utils/ops/bias_act.cpp -------------------------------------------------------------------------------- /torch_utils/ops/bias_act.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/torch_utils/ops/bias_act.cu -------------------------------------------------------------------------------- /torch_utils/ops/bias_act.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/torch_utils/ops/bias_act.h -------------------------------------------------------------------------------- /torch_utils/ops/bias_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/torch_utils/ops/bias_act.py -------------------------------------------------------------------------------- /torch_utils/ops/conv2d_gradfix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/torch_utils/ops/conv2d_gradfix.py -------------------------------------------------------------------------------- /torch_utils/ops/conv2d_resample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/torch_utils/ops/conv2d_resample.py -------------------------------------------------------------------------------- /torch_utils/ops/filtered_lrelu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/torch_utils/ops/filtered_lrelu.cpp -------------------------------------------------------------------------------- /torch_utils/ops/filtered_lrelu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/torch_utils/ops/filtered_lrelu.cu -------------------------------------------------------------------------------- /torch_utils/ops/filtered_lrelu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/torch_utils/ops/filtered_lrelu.h -------------------------------------------------------------------------------- /torch_utils/ops/filtered_lrelu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/torch_utils/ops/filtered_lrelu.py -------------------------------------------------------------------------------- /torch_utils/ops/filtered_lrelu_ns.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/torch_utils/ops/filtered_lrelu_ns.cu -------------------------------------------------------------------------------- /torch_utils/ops/filtered_lrelu_rd.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/torch_utils/ops/filtered_lrelu_rd.cu -------------------------------------------------------------------------------- /torch_utils/ops/filtered_lrelu_wr.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/torch_utils/ops/filtered_lrelu_wr.cu -------------------------------------------------------------------------------- /torch_utils/ops/fma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/torch_utils/ops/fma.py -------------------------------------------------------------------------------- /torch_utils/ops/grid_sample_gradfix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/torch_utils/ops/grid_sample_gradfix.py -------------------------------------------------------------------------------- /torch_utils/ops/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/torch_utils/ops/upfirdn2d.cpp -------------------------------------------------------------------------------- /torch_utils/ops/upfirdn2d.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/torch_utils/ops/upfirdn2d.cu -------------------------------------------------------------------------------- /torch_utils/ops/upfirdn2d.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/torch_utils/ops/upfirdn2d.h -------------------------------------------------------------------------------- /torch_utils/ops/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/torch_utils/ops/upfirdn2d.py -------------------------------------------------------------------------------- /torch_utils/persistence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/torch_utils/persistence.py -------------------------------------------------------------------------------- /torch_utils/training_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/torch_utils/training_stats.py -------------------------------------------------------------------------------- /torch_utils/utils_spectrum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/torch_utils/utils_spectrum.py -------------------------------------------------------------------------------- /trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/trainer.py -------------------------------------------------------------------------------- /ui/FaceTool.ui: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/ui/FaceTool.ui -------------------------------------------------------------------------------- /ui/ui_FaceTool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/ui/ui_FaceTool.py -------------------------------------------------------------------------------- /uic/FaceTool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boyzwj/zface/HEAD/uic/FaceTool.py --------------------------------------------------------------------------------