jones/facefusion
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Rename arcface_128_v2 to arcface_128

Browse Source
This commit is contained in:
henryruhs
2025-04-24 13:04:11 +02:00
parent 72a0edb6ba
commit 5118c7229f
6 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
facefusion/face_helper.py
View File

@@ -25,7 +25,7 @@ WARP_TEMPLATE_SET : WarpTemplateSet =\
[ 0.37097589, 0.82469196 ], [ 0.37097589, 0.82469196 ],
[ 0.63151696, 0.82325089 ] [ 0.63151696, 0.82325089 ]
]), ]),
'arcface_128_v2': numpy.array( 'arcface_128': numpy.array(
[ [
[ 0.36167656, 0.40387734 ], [ 0.36167656, 0.40387734 ],
[ 0.63696719, 0.40235469 ], [ 0.63696719, 0.40235469 ],

2
facefusion/processors/modules/expression_restorer.py
View File

@@ -68,7 +68,7 @@ def create_static_model_set(download_scope : DownloadScope) -> ModelSet:
'path': resolve_relative_path('../.assets/models/live_portrait_generator.onnx') 'path': resolve_relative_path('../.assets/models/live_portrait_generator.onnx')
} }
}, },
'template': 'arcface_128_v2', 'template': 'arcface_128',
'size': (512, 512) 'size': (512, 512)
} }
} }

2
facefusion/processors/modules/face_debugger.py
View File

@@ -90,7 +90,7 @@ def debug_face(target_face : Face, temp_vision_frame : VisionFrame) -> VisionFra
cv2.line(temp_vision_frame, (x1, y1), (x1, y2), primary_light_color, 3) cv2.line(temp_vision_frame, (x1, y1), (x1, y2), primary_light_color, 3)
if 'face-mask' in face_debugger_items: if 'face-mask' in face_debugger_items:
crop_vision_frame, affine_matrix = warp_face_by_face_landmark_5(temp_vision_frame, target_face.landmark_set.get('5/68'), 'arcface_128_v2', (512, 512)) crop_vision_frame, affine_matrix = warp_face_by_face_landmark_5(temp_vision_frame, target_face.landmark_set.get('5/68'), 'arcface_128', (512, 512))
inverse_matrix = cv2.invertAffineTransform(affine_matrix) inverse_matrix = cv2.invertAffineTransform(affine_matrix)
temp_size = temp_vision_frame.shape[:2][::-1] temp_size = temp_vision_frame.shape[:2][::-1]
crop_masks = [] crop_masks = []

2
facefusion/processors/modules/face_enhancer.py
View File

@@ -131,7 +131,7 @@ def create_static_model_set(download_scope : DownloadScope) -> ModelSet:
'path': resolve_relative_path('../.assets/models/gpen_bfr_256.onnx') 'path': resolve_relative_path('../.assets/models/gpen_bfr_256.onnx')
} }
}, },
'template': 'arcface_128_v2', 'template': 'arcface_128',
'size': (256, 256) 'size': (256, 256)
}, },
'gpen_bfr_512': 'gpen_bfr_512':

4
facefusion/processors/modules/face_swapper.py
View File

@@ -211,7 +211,7 @@ def create_static_model_set(download_scope : DownloadScope) -> ModelSet:
} }
}, },
'type': 'inswapper', 'type': 'inswapper',
'template': 'arcface_128_v2', 'template': 'arcface_128',
'size': (128, 128), 'size': (128, 128),
'mean': [ 0.0, 0.0, 0.0 ], 'mean': [ 0.0, 0.0, 0.0 ],
'standard_deviation': [ 1.0, 1.0, 1.0 ] 'standard_deviation': [ 1.0, 1.0, 1.0 ]
@@ -235,7 +235,7 @@ def create_static_model_set(download_scope : DownloadScope) -> ModelSet:
} }
}, },
'type': 'inswapper', 'type': 'inswapper',
'template': 'arcface_128_v2', 'template': 'arcface_128',
'size': (128, 128), 'size': (128, 128),
'mean': [ 0.0, 0.0, 0.0 ], 'mean': [ 0.0, 0.0, 0.0 ],
'standard_deviation': [ 1.0, 1.0, 1.0 ] 'standard_deviation': [ 1.0, 1.0, 1.0 ]

2
facefusion/types.py
View File

@@ -87,7 +87,7 @@ Content : TypeAlias = Dict[str, Any]
Commands : TypeAlias = List[str] Commands : TypeAlias = List[str]
WarpTemplate = Literal['arcface_112_v1', 'arcface_112_v2', 'arcface_128_v2', 'dfl_whole_face', 'ffhq_512', 'mtcnn_512', 'styleganex_384'] WarpTemplate = Literal['arcface_112_v1', 'arcface_112_v2', 'arcface_128', 'dfl_whole_face', 'ffhq_512', 'mtcnn_512', 'styleganex_384']
WarpTemplateSet : TypeAlias = Dict[WarpTemplate, NDArray[Any]] WarpTemplateSet : TypeAlias = Dict[WarpTemplate, NDArray[Any]]
ProcessMode = Literal['output', 'preview', 'stream'] ProcessMode = Literal['output', 'preview', 'stream']