From 8443f3512ade95ea1858efb4ea1e29400f7a1f28 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Mon, 3 Mar 2025 08:46:20 +0100 Subject: [PATCH] torch.float32 is the default --- face_swapper/src/helper.py | 28 ++++++---------------------- 1 file changed, 6 insertions(+), 22 deletions(-) diff --git a/face_swapper/src/helper.py b/face_swapper/src/helper.py index 1916588..765fe6a 100644 --- a/face_swapper/src/helper.py +++ b/face_swapper/src/helper.py @@ -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 ] + ]) }