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

Alphabetical order for warp templates

Browse Source
This commit is contained in:
henryruhs
2025-03-23 16:21:40 +01:00
parent 00d5c1f200
commit 4ebdeee634
2 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
face_swapper/src/helper.py
+6 -6
View File
@@ -5,20 +5,20 @@ from .types import EmbedderModule, Embedding, Mask, Padding, WarpTemplate, WarpT
WARP_TEMPLATE_SET : WarpTemplateSet =\
{
'vgg_face_hq_to_arcface_128_v2': torch.tensor(
'arcface_128_v2_to_arcface_112_v2': torch.tensor(
[
[ 1.01305414, -0.00140513, -0.00585911 ],
[ 0.00140513, 1.01305414, 0.11169602 ]
[ 8.75000016e-01, -1.07193451e-08, 3.80446920e-10 ],
[ 1.07193451e-08, 8.75000016e-01, -1.25000007e-01 ]
]),
'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(
'vgg_face_hq_to_arcface_128_v2': torch.tensor(
[
[ 8.75000016e-01, -1.07193451e-08, 3.80446920e-10 ],
[ 1.07193451e-08, 8.75000016e-01, -1.25000007e-01 ]
[ 1.01305414, -0.00140513, -0.00585911 ],
[ 0.00140513, 1.01305414, 0.11169602 ]
])
}
face_swapper/src/types.py
+1 -1
View File
@@ -21,5 +21,5 @@ FaceMaskerModule : TypeAlias = Module
OptimizerSet : TypeAlias = Any
WarpTemplate = Literal['vgg_face_hq_to_arcface_128_v2', 'ffhq_to_arcface_128_v2', 'arcface_128_v2_to_arcface_112_v2']
WarpTemplate = Literal['arcface_128_v2_to_arcface_112_v2', 'ffhq_to_arcface_128_v2', 'vgg_face_hq_to_arcface_128_v2']
WarpTemplateSet : TypeAlias = Dict[WarpTemplate, Tensor]