diff --git a/core/base_landmarks.py b/core/base_mesh.py similarity index 99% rename from core/base_landmarks.py rename to core/base_mesh.py index d049bd1..e0b128a 100644 --- a/core/base_landmarks.py +++ b/core/base_mesh.py @@ -31,7 +31,7 @@ class MediapipeBaseLandmarks: # Calculate horizontal distance from center (normalized to 0-1) dx = np.abs(transformed[:, 0] - center_x) - influence = np.clip(dx / center_x, 0, 1) # 0 w środku, 1 na krawędziach + influence = np.clip(dx / center_x, 0, 1) # Calculate scaled positions with horizontal influence scale_factor = 1.0 + (x_scale - 1.0) * influence diff --git a/core/cpu_deformer.py b/core/cpu_deformer.py index 6895a3f..0f0ecb1 100644 --- a/core/cpu_deformer.py +++ b/core/cpu_deformer.py @@ -1,6 +1,6 @@ import numpy as np from PIL import Image -from ..core.base_landmarks import MediapipeBaseLandmarks +from ..core.base_mesh import MediapipeBaseLandmarks class CPUDeformer: """ diff --git a/core/gpu_deformer.py b/core/gpu_deformer.py index 53f73d9..c2c71b2 100644 --- a/core/gpu_deformer.py +++ b/core/gpu_deformer.py @@ -1,7 +1,7 @@ import numpy as np from PIL import Image import torch -from ..core.base_landmarks import MediapipeBaseLandmarks +from ..core.base_mesh import MediapipeBaseLandmarks class GPUDeformer: diff --git a/nodes/face_wrapper.py b/nodes/face_wrapper.py index abb9035..0ded79f 100644 --- a/nodes/face_wrapper.py +++ b/nodes/face_wrapper.py @@ -6,7 +6,7 @@ import cv2 from ..core.face_detector import FaceDetector from ..core.image_processor import ImageProcessor -from ..core.base_landmarks import MediapipeBaseLandmarks +from ..core.base_mesh import MediapipeBaseLandmarks from ..core.cpu_deformer import CPUDeformer from ..core.gpu_deformer import GPUDeformer diff --git a/nodes/tools.py b/nodes/tools.py index a198eec..1eb6d4c 100644 --- a/nodes/tools.py +++ b/nodes/tools.py @@ -1,7 +1,6 @@ import torch import numpy as np import cv2 -from typing import Tuple from torch import Tensor