FaceTracker init

This commit is contained in:
Rafał Tłuchowski
2025-02-23 19:46:59 +01:00
parent d5018b928f
commit 9e1d91cd5d
2 changed files with 31 additions and 3 deletions
+6 -3
View File
@@ -4,6 +4,7 @@ from .nodes.image_feeder import ImageFeeder
from .nodes.image_filters import HighPassFilter
from .nodes.face_wrapper import FaceWrapper
from .nodes.face_fit_and_restore import FaceFitAndRestore
from .nodes.face_tracker import FaceTracker
# Get the path to the current directory
NODE_PATH = os.path.dirname(os.path.realpath(__file__))
@@ -12,14 +13,16 @@ NODE_CLASS_MAPPINGS = {
"FaceFitAndRestore": FaceFitAndRestore,
"FaceWrapper": FaceWrapper,
"HighPassFilter": HighPassFilter,
"ImageFeeder": ImageFeeder
"ImageFeeder": ImageFeeder,
"FaceTracker": FaceTracker
}
NODE_DISPLAY_NAME_MAPPINGS = {
"FaceFitAndRestore": "Face Fit or Restore",
"FaceWrapper": "Face Wrapper",
"HighPassFilter": "High Pass Filter (HPF)",
"ImageFeeder": "Image Feeder"
"ImageFeeder": "Image Feeder",
"FaceTracker": "Face Tracker"
}
__version__ = "1.1.5"
__version__ = "1.2.0"
+25
View File
@@ -0,0 +1,25 @@
from ..core.face_detector import FaceDetector
from ..core.image_processor import ImageProcessor
class FaceTracker:
"""ComfyUI node for tracking facial features across image sequences."""
def __init__(self):
self.face_detector = FaceDetector()
self.image_processor = ImageProcessor()
@classmethod
def INPUT_TYPES(cls):
return {
"required": {
"image": ("IMAGE",),
}
}
RETURN_TYPES = ("IMAGE",)
FUNCTION = "track_face"
CATEGORY = "Face Processor"
def track_face(self, image):
return (image,)