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

torch.float32 is the default

Browse Source
This commit is contained in:
henryruhs
2025-03-03 08:46:20 +01:00
parent d3a2035d7a
commit 8443f3512a
1 changed files with 6 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files
face_swapper/src/helper.py
+6 -22
View File
@@ -7,30 +7,14 @@ WARP_MATRIX_SET : WarpMatrixSet =\
{
'vgg_face_hq_to_arcface_128_v2': torch.tensor(
[
[
1.01305414,
-0.00140513,
-0.00585911
],
[
0.00140513,
1.01305414,
0.11169602
]
], dtype = torch.float32),
[ 1.01305414, -0.00140513, -0.00585911 ],
[ 0.00140513, 1.01305414, 0.11169602 ]
]),
'arcface_128_v2_to_arcface_112_v2': torch.tensor(
[
[
8.75000016e-01,
-1.07193451e-08,
3.80446920e-10
],
[
1.07193451e-08,
8.75000016e-01,
-1.25000007e-01
]
], dtype = torch.float32)
[ 8.75000016e-01, -1.07193451e-08, 3.80446920e-10 ],
[ 1.07193451e-08, 8.75000016e-01, -1.25000007e-01 ]
])
}