diff --git a/nodes.py b/nodes.py index e7206f7..9e79258 100644 --- a/nodes.py +++ b/nodes.py @@ -11,7 +11,7 @@ def deepface_image_from_comfy_image(comfy_image): image_data = np.clip(255 * comfy_image.cpu().numpy(), 0, 255).astype(np.uint8) return image_data[:, :, ::-1] # Convert RGB to BGR -class DeepfacePrepareNode: +class DeepfaceExtractFacesNode: def __init__(self): pass @@ -68,7 +68,7 @@ class DeepfaceVerifyNode: "display": "number", "minimum": 0.0, "maximum": 1.0, - "step": 0.1, + "step": 0.01, }) }, } @@ -78,8 +78,6 @@ class DeepfaceVerifyNode: FUNCTION = "run" - # OUTPUT_NODE = False - CATEGORY = "deepface/verify" def run(self, images, face_images, threshold): @@ -118,11 +116,11 @@ class DeepfaceVerifyNode: return (torch.stack(output_images, dim=0),) NODE_CLASS_MAPPINGS = { - "DeepfacePrepare": DeepfacePrepareNode, + "DeepfaceExtractFaces": DeepfaceExtractFacesNode, "DeepfaceVerify": DeepfaceVerifyNode, } NODE_DISPLAY_NAME_MAPPINGS = { - "DeepfacePrepare": "Deepface Prepare", + "DeepfaceExtractFaces": "Deepface Extract Faces", "DeepfaceVerify": "Deepface Verify", }