├── .gitignore ├── LICENSE ├── README.md ├── checkpoints └── BFM │ └── std_exp.txt ├── compare_d3dfr.py ├── demo_face_restoration.py ├── environment.yml ├── examples ├── main_image.png ├── restoration │ ├── 0007.jpg │ ├── 0022.jpg │ ├── 0026.jpg │ ├── 0067.jpg │ ├── 0149.jpg │ ├── 0327.jpg │ ├── 0357.jpg │ ├── 0437.jpg │ ├── 0525.jpg │ ├── test.jpg │ ├── test2.jpg │ └── test3.jpg ├── test │ ├── 0007.jpg │ ├── 0022.jpg │ ├── 0026.jpg │ ├── 0067.jpg │ ├── 0149.jpg │ ├── 0327.jpg │ ├── 0357.jpg │ ├── 0437.jpg │ ├── 0525.jpg │ ├── test2.jpg │ └── test3.jpg └── test_crop │ └── test.jpg ├── face_d3dfr ├── BFM09Model.py ├── BaseModel.py ├── D3DFR_ReconModel.py ├── __init__.py ├── d3dfr_pytorch.py └── preprocess.py ├── face_detection └── yolov5_face.py ├── face_restoration ├── model_sgpn.py └── op │ ├── __init__.py │ ├── fused_act.py │ ├── fused_bias_act.cpp │ ├── fused_bias_act_kernel.cu │ ├── upfirdn2d.cpp │ ├── upfirdn2d.py │ └── upfirdn2d_kernel.cu └── utils ├── __init__.py ├── cropface.py └── umeyama.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/README.md -------------------------------------------------------------------------------- /checkpoints/BFM/std_exp.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/checkpoints/BFM/std_exp.txt -------------------------------------------------------------------------------- /compare_d3dfr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/compare_d3dfr.py -------------------------------------------------------------------------------- /demo_face_restoration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/demo_face_restoration.py -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/environment.yml -------------------------------------------------------------------------------- /examples/main_image.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/examples/main_image.png -------------------------------------------------------------------------------- /examples/restoration/0007.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/examples/restoration/0007.jpg -------------------------------------------------------------------------------- /examples/restoration/0022.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/examples/restoration/0022.jpg -------------------------------------------------------------------------------- /examples/restoration/0026.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/examples/restoration/0026.jpg -------------------------------------------------------------------------------- /examples/restoration/0067.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/examples/restoration/0067.jpg -------------------------------------------------------------------------------- /examples/restoration/0149.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/examples/restoration/0149.jpg -------------------------------------------------------------------------------- /examples/restoration/0327.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/examples/restoration/0327.jpg -------------------------------------------------------------------------------- /examples/restoration/0357.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/examples/restoration/0357.jpg -------------------------------------------------------------------------------- /examples/restoration/0437.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/examples/restoration/0437.jpg -------------------------------------------------------------------------------- /examples/restoration/0525.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/examples/restoration/0525.jpg -------------------------------------------------------------------------------- /examples/restoration/test.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/examples/restoration/test.jpg -------------------------------------------------------------------------------- /examples/restoration/test2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/examples/restoration/test2.jpg -------------------------------------------------------------------------------- /examples/restoration/test3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/examples/restoration/test3.jpg -------------------------------------------------------------------------------- /examples/test/0007.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/examples/test/0007.jpg -------------------------------------------------------------------------------- /examples/test/0022.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/examples/test/0022.jpg -------------------------------------------------------------------------------- /examples/test/0026.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/examples/test/0026.jpg -------------------------------------------------------------------------------- /examples/test/0067.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/examples/test/0067.jpg -------------------------------------------------------------------------------- /examples/test/0149.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/examples/test/0149.jpg -------------------------------------------------------------------------------- /examples/test/0327.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/examples/test/0327.jpg -------------------------------------------------------------------------------- /examples/test/0357.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/examples/test/0357.jpg -------------------------------------------------------------------------------- /examples/test/0437.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/examples/test/0437.jpg -------------------------------------------------------------------------------- /examples/test/0525.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/examples/test/0525.jpg -------------------------------------------------------------------------------- /examples/test/test2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/examples/test/test2.jpg -------------------------------------------------------------------------------- /examples/test/test3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/examples/test/test3.jpg -------------------------------------------------------------------------------- /examples/test_crop/test.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/examples/test_crop/test.jpg -------------------------------------------------------------------------------- /face_d3dfr/BFM09Model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/face_d3dfr/BFM09Model.py -------------------------------------------------------------------------------- /face_d3dfr/BaseModel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/face_d3dfr/BaseModel.py -------------------------------------------------------------------------------- /face_d3dfr/D3DFR_ReconModel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/face_d3dfr/D3DFR_ReconModel.py -------------------------------------------------------------------------------- /face_d3dfr/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /face_d3dfr/d3dfr_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/face_d3dfr/d3dfr_pytorch.py -------------------------------------------------------------------------------- /face_d3dfr/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/face_d3dfr/preprocess.py -------------------------------------------------------------------------------- /face_detection/yolov5_face.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/face_detection/yolov5_face.py -------------------------------------------------------------------------------- /face_restoration/model_sgpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/face_restoration/model_sgpn.py -------------------------------------------------------------------------------- /face_restoration/op/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/face_restoration/op/__init__.py -------------------------------------------------------------------------------- /face_restoration/op/fused_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/face_restoration/op/fused_act.py -------------------------------------------------------------------------------- /face_restoration/op/fused_bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/face_restoration/op/fused_bias_act.cpp -------------------------------------------------------------------------------- /face_restoration/op/fused_bias_act_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/face_restoration/op/fused_bias_act_kernel.cu -------------------------------------------------------------------------------- /face_restoration/op/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/face_restoration/op/upfirdn2d.cpp -------------------------------------------------------------------------------- /face_restoration/op/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/face_restoration/op/upfirdn2d.py -------------------------------------------------------------------------------- /face_restoration/op/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/face_restoration/op/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/cropface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/utils/cropface.py -------------------------------------------------------------------------------- /utils/umeyama.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/FaceRestoration-sgpn/HEAD/utils/umeyama.py --------------------------------------------------------------------------------