├── IMGS └── 061400.png ├── LICENSE ├── README.md ├── __init_paths.py ├── align_faces.py ├── distributed.py ├── environment.yaml ├── face_model ├── arcface.py ├── gpen_model.py └── op │ ├── __init__.py │ ├── __pycache__ │ ├── __init__.cpython-39.pyc │ ├── fused_act.cpython-39.pyc │ └── upfirdn2d.cpython-39.pyc │ ├── fused_act.py │ ├── fused_bias_act.cpp │ ├── fused_bias_act_kernel.cu │ ├── upfirdn2d.cpp │ ├── upfirdn2d.py │ └── upfirdn2d_kernel.cu ├── face_parsing ├── bisenet.py ├── face_parsing.py └── resnet.py ├── misc ├── cog.yaml ├── onnx_export.py └── predict.py ├── saved_models └── README.md ├── train_simple.py └── training ├── data_loader ├── dataset_face.py └── degradations.py ├── loss ├── helpers.py ├── id_loss.py └── model_irse.py └── lpips ├── __init__.py ├── lpips.py ├── pretrained_networks.py ├── trainer.py └── weights ├── v0.0 ├── alex.pth ├── squeeze.pth └── vgg.pth └── v0.1 ├── alex.pth ├── squeeze.pth └── vgg.pth /IMGS/061400.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/IMGS/061400.png -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/README.md -------------------------------------------------------------------------------- /__init_paths.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/__init_paths.py -------------------------------------------------------------------------------- /align_faces.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/align_faces.py -------------------------------------------------------------------------------- /distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/distributed.py -------------------------------------------------------------------------------- /environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/environment.yaml -------------------------------------------------------------------------------- /face_model/arcface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/face_model/arcface.py -------------------------------------------------------------------------------- /face_model/gpen_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/face_model/gpen_model.py -------------------------------------------------------------------------------- /face_model/op/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/face_model/op/__init__.py -------------------------------------------------------------------------------- /face_model/op/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/face_model/op/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /face_model/op/__pycache__/fused_act.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/face_model/op/__pycache__/fused_act.cpython-39.pyc -------------------------------------------------------------------------------- /face_model/op/__pycache__/upfirdn2d.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/face_model/op/__pycache__/upfirdn2d.cpython-39.pyc -------------------------------------------------------------------------------- /face_model/op/fused_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/face_model/op/fused_act.py -------------------------------------------------------------------------------- /face_model/op/fused_bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/face_model/op/fused_bias_act.cpp -------------------------------------------------------------------------------- /face_model/op/fused_bias_act_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/face_model/op/fused_bias_act_kernel.cu -------------------------------------------------------------------------------- /face_model/op/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/face_model/op/upfirdn2d.cpp -------------------------------------------------------------------------------- /face_model/op/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/face_model/op/upfirdn2d.py -------------------------------------------------------------------------------- /face_model/op/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/face_model/op/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /face_parsing/bisenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/face_parsing/bisenet.py -------------------------------------------------------------------------------- /face_parsing/face_parsing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/face_parsing/face_parsing.py -------------------------------------------------------------------------------- /face_parsing/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/face_parsing/resnet.py -------------------------------------------------------------------------------- /misc/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/misc/cog.yaml -------------------------------------------------------------------------------- /misc/onnx_export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/misc/onnx_export.py -------------------------------------------------------------------------------- /misc/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/misc/predict.py -------------------------------------------------------------------------------- /saved_models/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/saved_models/README.md -------------------------------------------------------------------------------- /train_simple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/train_simple.py -------------------------------------------------------------------------------- /training/data_loader/dataset_face.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/training/data_loader/dataset_face.py -------------------------------------------------------------------------------- /training/data_loader/degradations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/training/data_loader/degradations.py -------------------------------------------------------------------------------- /training/loss/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/training/loss/helpers.py -------------------------------------------------------------------------------- /training/loss/id_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/training/loss/id_loss.py -------------------------------------------------------------------------------- /training/loss/model_irse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/training/loss/model_irse.py -------------------------------------------------------------------------------- /training/lpips/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/training/lpips/__init__.py -------------------------------------------------------------------------------- /training/lpips/lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/training/lpips/lpips.py -------------------------------------------------------------------------------- /training/lpips/pretrained_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/training/lpips/pretrained_networks.py -------------------------------------------------------------------------------- /training/lpips/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/training/lpips/trainer.py -------------------------------------------------------------------------------- /training/lpips/weights/v0.0/alex.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/training/lpips/weights/v0.0/alex.pth -------------------------------------------------------------------------------- /training/lpips/weights/v0.0/squeeze.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/training/lpips/weights/v0.0/squeeze.pth -------------------------------------------------------------------------------- /training/lpips/weights/v0.0/vgg.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/training/lpips/weights/v0.0/vgg.pth -------------------------------------------------------------------------------- /training/lpips/weights/v0.1/alex.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/training/lpips/weights/v0.1/alex.pth -------------------------------------------------------------------------------- /training/lpips/weights/v0.1/squeeze.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/training/lpips/weights/v0.1/squeeze.pth -------------------------------------------------------------------------------- /training/lpips/weights/v0.1/vgg.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MingtaoGuo/StyleSwap/HEAD/training/lpips/weights/v0.1/vgg.pth --------------------------------------------------------------------------------