From 53641e1779bef768bf0caa9936660b2275897cb9 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Wed, 11 Jun 2025 15:11:19 +0200 Subject: [PATCH] Rename head to face --- facefusion/choices.py | 4 ++-- facefusion/types.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/facefusion/choices.py b/facefusion/choices.py index 726edc3..24e2e6b 100755 --- a/facefusion/choices.py +++ b/facefusion/choices.py @@ -22,8 +22,8 @@ face_parser_models : List[FaceParserModel] = [ 'bisenet_resnet_18', 'bisenet_res face_mask_types : List[FaceMaskType] = [ 'box', 'occlusion', 'area', 'region' ] face_mask_area_set : FaceMaskAreaSet =\ { - 'upper-head': [ 2, 14, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47 ], - 'lower-head': [ 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67 ], + 'upper-face': [ 2, 14, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47 ], + 'lower-face': [ 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67 ], 'mouth': [ 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67 ] } face_mask_region_set : FaceMaskRegionSet =\ diff --git a/facefusion/types.py b/facefusion/types.py index ee3fc76..ca32f30 100755 --- a/facefusion/types.py +++ b/facefusion/types.py @@ -108,7 +108,7 @@ FaceSelectorOrder = Literal['left-right', 'right-left', 'top-bottom', 'bottom-to FaceOccluderModel = Literal['xseg_1', 'xseg_2', 'xseg_3'] FaceParserModel = Literal['bisenet_resnet_18', 'bisenet_resnet_34'] FaceMaskType = Literal['box', 'occlusion', 'area', 'region'] -FaceMaskArea = Literal['upper-head', 'lower-head', 'mouth'] +FaceMaskArea = Literal['upper-face', 'lower-face', 'mouth'] FaceMaskRegion = Literal['skin', 'left-eyebrow', 'right-eyebrow', 'left-eye', 'right-eye', 'glasses', 'nose', 'mouth', 'upper-lip', 'lower-lip'] FaceMaskRegionSet : TypeAlias = Dict[FaceMaskRegion, int] FaceMaskAreaSet : TypeAlias = Dict[FaceMaskArea, List[int]]