update node name
This commit is contained in:
+2
-2
@@ -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
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user