CalvinBackup/facefusion-labs
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion-labs.git synced 2026-05-22 23:59:40 +02:00
Code Issues Packages Projects Releases Wiki Activity

add 'ffhq_to_arcface_128_v2' template

Browse Source
This commit is contained in:
harisreedhar
2025-03-22 14:14:37 +05:30
parent 35afb426b7
commit 9153b4ce8f
2 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
face_swapper/src/helper.py
+5
View File
@@ -10,6 +10,11 @@ WARP_TEMPLATE_SET : WarpTemplateSet =\
[ 1.01305414, -0.00140513, -0.00585911 ],
[ 0.00140513, 1.01305414, 0.11169602 ]
]),
'ffhq_to_arcface_128_v2': torch.tensor(
[
[ 8.50048894e-01, -1.29486822e-04, 1.90956388e-03 ],
[ 1.29486822e-04, 8.50048894e-01, 9.56254653e-02 ]
]),
'arcface_128_v2_to_arcface_112_v2': torch.tensor(
[
[ 8.75000016e-01, -1.07193451e-08, 3.80446920e-10 ],
face_swapper/src/types.py
+1 -1
View File
@@ -21,5 +21,5 @@ FaceParserModule : TypeAlias = Module
OptimizerSet : TypeAlias = Any
WarpTemplate = Literal['vgg_face_hq_to_arcface_128_v2', 'arcface_128_v2_to_arcface_112_v2']
WarpTemplate = Literal['vgg_face_hq_to_arcface_128_v2', 'ffhq_to_arcface_128_v2', 'arcface_128_v2_to_arcface_112_v2']
WarpTemplateSet : TypeAlias = Dict[WarpTemplate, Tensor]