From d96feac94e76f8a28727c2f00e20401f83ac2510 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Rafa=C5=82=20T=C5=82uchowski?= Date: Mon, 27 Jan 2025 21:24:21 +0100 Subject: [PATCH] new tool - High Pass Filter (HPF) --- core/{base_landmarks.py => base_mesh.py} | 2 +- core/cpu_deformer.py | 2 +- core/gpu_deformer.py | 2 +- nodes/face_wrapper.py | 2 +- nodes/tools.py | 1 - 5 files changed, 4 insertions(+), 5 deletions(-) rename core/{base_landmarks.py => base_mesh.py} (99%) 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