CalvinBackup/facefusion
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion.git synced 2026-06-03 03:08:01 +02:00
Code Issues Packages Projects Releases Wiki Activity

Minor cleanups

Browse Source
This commit is contained in:
henryruhs
2024-01-30 21:26:24 +01:00
parent ce579846b4
commit 0b345a7e33
4 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
facefusion/choices.py
+1 -1
View File
@@ -6,7 +6,7 @@ from facefusion.common_helper import create_int_range, create_float_range
video_memory_strategies : List[VideoMemoryStrategy] = [ 'strict', 'moderate', 'tolerant' ]
face_analyser_orders : List[FaceAnalyserOrder] = [ 'left-right', 'right-left', 'top-bottom', 'bottom-top', 'small-large', 'large-small', 'best-worst', 'worst-best' ]
face_analyser_ages : List[FaceAnalyserAge] = [ 'child', 'teen', 'adult', 'senior' ]
face_analyser_genders : List[FaceAnalyserGender] = [ 'male', 'female' ]
face_analyser_genders : List[FaceAnalyserGender] = [ 'female', 'male' ]
face_detector_models : List[FaceDetectorModel] = [ 'retinaface', 'yoloface', 'yunet' ]
face_detector_sizes : List[str] = [ '160x160', '320x320', '480x480', '512x512', '640x640', '768x768', '960x960', '1024x1024' ]
face_selector_modes : List[FaceSelectorMode] = [ 'reference', 'one', 'many' ]
facefusion/processors/frame/modules/face_debugger.py
+1 -1
View File
@@ -98,7 +98,7 @@ def debug_face(source_face : Face, target_face : Face, reference_faces : FaceSet
cv2.circle(temp_frame, (kps[index][0], kps[index][1]), 3, primary_color, -1)
if 'score' in frame_processors_globals.face_debugger_items:
face_score_text = str(round(target_face.score, 2))
face_score_position = (bounding_box[0] + 10, bounding_box[1] + 20)
face_score_position = (bounding_box[0] + 20, bounding_box[1] + 20)
cv2.putText(temp_frame, face_score_text, face_score_position, cv2.FONT_HERSHEY_SIMPLEX, 0.5, secondary_color, 2)
return temp_frame
facefusion/processors/frame/typings.py
+1 -1
View File
@@ -1,6 +1,6 @@
from typing import Literal
FaceDebuggerItem = Literal['bbox', 'kps', 'face-mask', 'score', 'distance']
FaceDebuggerItem = Literal['bbox', 'kps', 'face-mask', 'score']
FaceEnhancerModel = Literal['codeformer', 'gfpgan_1.2', 'gfpgan_1.3', 'gfpgan_1.4', 'gpen_bfr_256', 'gpen_bfr_512', 'restoreformer_plus_plus']
FaceSwapperModel = Literal['blendswap_256', 'inswapper_128', 'inswapper_128_fp16', 'simswap_256', 'simswap_512_unofficial']
FrameEnhancerModel = Literal['real_esrgan_x2plus', 'real_esrgan_x4plus', 'real_esrnet_x4plus']
facefusion/typing.py
+1 -1
View File
@@ -46,7 +46,7 @@ VideoMemoryStrategy = Literal['strict', 'moderate', 'tolerant']
FaceSelectorMode = Literal['reference', 'one', 'many']
FaceAnalyserOrder = Literal['left-right', 'right-left', 'top-bottom', 'bottom-top', 'small-large', 'large-small', 'best-worst', 'worst-best']
FaceAnalyserAge = Literal['child', 'teen', 'adult', 'senior']
FaceAnalyserGender = Literal['male', 'female']
FaceAnalyserGender = Literal['female', 'male']
FaceDetectorModel = Literal['retinaface', 'yoloface', 'yunet']
FaceRecognizerModel = Literal['arcface_blendswap', 'arcface_inswapper', 'arcface_simswap']
FaceMaskType = Literal['box', 'occlusion', 'region']