Rename node

This commit is contained in:
Jordan Phillips
2024-02-14 13:10:55 -08:00
parent 53df581859
commit ff6fdb20f4
+4 -6
View File
@@ -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",
}