update node name

This commit is contained in:
toto
2023-11-22 17:15:09 +08:00
parent a78fce7be5
commit 5c0ef3e589
2 changed files with 4 additions and 4 deletions
+2 -2
View File
@@ -17,7 +17,7 @@ NODE_CLASS_MAPPINGS = {
"FC_Segment": FCSegment,
"FC_ReplaceImage": FCReplaceImage,
"FC_CropBottom": FCCropBottom,
"FC_FCCropFace": FCCropFace,
"FC_CropFace": FCCropFace,
}
NODE_DISPLAY_NAME_MAPPINGS = {
"FC_FaceFusion": "FC FaceFusion",
@@ -26,7 +26,7 @@ NODE_DISPLAY_NAME_MAPPINGS = {
"FC_CropMask": "FC CropMask",
"FC_ReplaceImage": "FC ReplaceImage",
"FC_CropBottom": "FC CropBottom",
"FC_FCCropFace": "FC FCCropFace",
"FC_CropFace": "FC CropFace",
}
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
+2 -2
View File
@@ -292,7 +292,7 @@ class FCCropFace:
}
}
RETURN_TYPES = ("IMAGE", "BOX")
RETURN_TYPES = ("IMAGE", "BOX", "KEY_POINT")
FUNCTION = "face_crop"
CATEGORY = "facechain/crop"
@@ -324,4 +324,4 @@ class FCCropFace:
bbox[3] = np.clip(np.array(bbox[3], np.int32) + face_h * (crop_ratio - 1) / 2, 0, h - 1)
bbox = np.array(bbox, np.int32)
result_image = source_image[:, bbox[3]:bbox[1], bbox[0]:bbox[2], :]
return result_image, bbox
return result_image, bbox , points_array