├── README.md ├── criteria ├── __init__.py ├── cx_style_loss.py └── id_loss.py ├── demo └── teaser.jpg ├── examples ├── img │ ├── 001.jpg │ ├── 002.jpg │ ├── 003.jpg │ ├── 004.jpg │ ├── 005.jpg │ └── 006.jpg ├── pair_reenact.txt ├── pair_swap.txt ├── reenact │ └── pair.txt └── swap │ └── pair.txt ├── generate_reenact.py ├── generate_swap.py ├── install.sh ├── train.py ├── train_reenact.py ├── train_swap.py ├── training ├── __init__.py ├── base_dataset.py ├── dataset.py ├── dataset_ddp.py ├── lpips │ ├── __init__.py │ ├── base_model.py │ ├── dist_model.py │ ├── networks_basic.py │ ├── pretrained_networks.py │ └── weights │ │ ├── v0.0 │ │ ├── alex.pth │ │ ├── squeeze.pth │ │ └── vgg.pth │ │ └── v0.1 │ │ ├── alex.pth │ │ ├── squeeze.pth │ │ └── vgg.pth ├── model.py ├── op │ ├── __init__.py │ ├── fused_act.py │ ├── fused_bias_act.cpp │ ├── fused_bias_act_kernel.cu │ ├── upfirdn2d.cpp │ ├── upfirdn2d.py │ └── upfirdn2d_kernel.cu ├── pose.py └── vgg.py └── utils ├── common.py └── flow_utils.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/README.md -------------------------------------------------------------------------------- /criteria/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /criteria/cx_style_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/criteria/cx_style_loss.py -------------------------------------------------------------------------------- /criteria/id_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/criteria/id_loss.py -------------------------------------------------------------------------------- /demo/teaser.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/demo/teaser.jpg -------------------------------------------------------------------------------- /examples/img/001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/examples/img/001.jpg -------------------------------------------------------------------------------- /examples/img/002.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/examples/img/002.jpg -------------------------------------------------------------------------------- /examples/img/003.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/examples/img/003.jpg -------------------------------------------------------------------------------- /examples/img/004.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/examples/img/004.jpg -------------------------------------------------------------------------------- /examples/img/005.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/examples/img/005.jpg -------------------------------------------------------------------------------- /examples/img/006.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/examples/img/006.jpg -------------------------------------------------------------------------------- /examples/pair_reenact.txt: -------------------------------------------------------------------------------- 1 | 001_002 2 | 003_004 -------------------------------------------------------------------------------- /examples/pair_swap.txt: -------------------------------------------------------------------------------- 1 | 006_005 -------------------------------------------------------------------------------- /examples/reenact/pair.txt: -------------------------------------------------------------------------------- 1 | 001_002 2 | 003_004 -------------------------------------------------------------------------------- /examples/swap/pair.txt: -------------------------------------------------------------------------------- 1 | 001_002 2 | 003_004 -------------------------------------------------------------------------------- /generate_reenact.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/generate_reenact.py -------------------------------------------------------------------------------- /generate_swap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/generate_swap.py -------------------------------------------------------------------------------- /install.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/install.sh -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/train.py -------------------------------------------------------------------------------- /train_reenact.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/train_reenact.py -------------------------------------------------------------------------------- /train_swap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/train_swap.py -------------------------------------------------------------------------------- /training/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /training/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/training/base_dataset.py -------------------------------------------------------------------------------- /training/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/training/dataset.py -------------------------------------------------------------------------------- /training/dataset_ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/training/dataset_ddp.py -------------------------------------------------------------------------------- /training/lpips/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/training/lpips/__init__.py -------------------------------------------------------------------------------- /training/lpips/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/training/lpips/base_model.py -------------------------------------------------------------------------------- /training/lpips/dist_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/training/lpips/dist_model.py -------------------------------------------------------------------------------- /training/lpips/networks_basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/training/lpips/networks_basic.py -------------------------------------------------------------------------------- /training/lpips/pretrained_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/training/lpips/pretrained_networks.py -------------------------------------------------------------------------------- /training/lpips/weights/v0.0/alex.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/training/lpips/weights/v0.0/alex.pth -------------------------------------------------------------------------------- /training/lpips/weights/v0.0/squeeze.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/training/lpips/weights/v0.0/squeeze.pth -------------------------------------------------------------------------------- /training/lpips/weights/v0.0/vgg.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/training/lpips/weights/v0.0/vgg.pth -------------------------------------------------------------------------------- /training/lpips/weights/v0.1/alex.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/training/lpips/weights/v0.1/alex.pth -------------------------------------------------------------------------------- /training/lpips/weights/v0.1/squeeze.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/training/lpips/weights/v0.1/squeeze.pth -------------------------------------------------------------------------------- /training/lpips/weights/v0.1/vgg.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/training/lpips/weights/v0.1/vgg.pth -------------------------------------------------------------------------------- /training/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/training/model.py -------------------------------------------------------------------------------- /training/op/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/training/op/__init__.py -------------------------------------------------------------------------------- /training/op/fused_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/training/op/fused_act.py -------------------------------------------------------------------------------- /training/op/fused_bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/training/op/fused_bias_act.cpp -------------------------------------------------------------------------------- /training/op/fused_bias_act_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/training/op/fused_bias_act_kernel.cu -------------------------------------------------------------------------------- /training/op/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/training/op/upfirdn2d.cpp -------------------------------------------------------------------------------- /training/op/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/training/op/upfirdn2d.py -------------------------------------------------------------------------------- /training/op/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/training/op/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /training/pose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/training/pose.py -------------------------------------------------------------------------------- /training/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/training/vgg.py -------------------------------------------------------------------------------- /utils/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/utils/common.py -------------------------------------------------------------------------------- /utils/flow_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xc-csc101/UniFace/HEAD/utils/flow_utils.py --------------------------------------------------------------------------------