From 1e248e60f79df186c0bea5e129d508701bd56efe Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Rafa=C5=82=20T=C5=82uchowski?= Date: Sun, 26 Jan 2025 13:06:12 +0100 Subject: [PATCH] WIP cpu and gpu deformer --- core/gpu_deformer.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/core/gpu_deformer.py b/core/gpu_deformer.py index 6517f21..b7e38ff 100644 --- a/core/gpu_deformer.py +++ b/core/gpu_deformer.py @@ -10,7 +10,7 @@ class GPUDeformer: """ @staticmethod - def warp_face(image, source_landmarks, target_landmarks, device='cuda'): + def warp_face(image, source_landmarks, target_landmarks): """ Warp the face image using parallel GPU processing with reverse mapping. Now includes boundary triangles for complete image warping. @@ -19,7 +19,6 @@ class GPUDeformer: image: Input PIL Image or numpy array source_landmarks: Detected face landmarks target_landmarks: Target base landmarks - device: 'cuda' or 'cpu' (default: 'cuda') Returns: PIL Image: Warped image