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

Add deep swapper models by Mats

Browse Source
This commit is contained in:
henryruhs 2024-11-18 22:26:31 +01:00
parent 3cf06de27f
commit b6e895fcf6
3 changed files with 86 additions and 44 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
facefusion/processors/choices.py
View File

@ -6,9 +6,6 @@ from facefusion.processors.typing import AgeModifierModel, DeepSwapperModel, Exp
age_modifier_models : List[AgeModifierModel] = [ 'styleganex_age' ]
deep_swapper_models : List[DeepSwapperModel] =\
[
'iperov/emma_watson_224',
'iperov/keanu_reeves_320',
'iperov/jackie_chan_224',
'iperov/alexandra_daddario_224',
'iperov/alexei_navalny_224',
'iperov/amber_heard_224',
@ -27,7 +24,7 @@ deep_swapper_models : List[DeepSwapperModel] =\
'iperov/meghan_markle_224',
'iperov/natalie_dormer_224',
'iperov/natalie_portman_224',
'iperov/nicolas_coppola__224',
'iperov/nicolas_coppola_224',
'iperov/robert_downey_224',
'iperov/rowan_atkinson_224',
'iperov/ryan_reynolds_224',
@ -37,7 +34,24 @@ deep_swapper_models : List[DeepSwapperModel] =\
'iperov/thomas_cruise_224',
'iperov/thomas_holland_224',
'iperov/vin_diesel_224',
'iperov/vladimir_putin_224'
'iperov/vladimir_putin_224',
'mats/alica_schmidt_320',
'mats/ashley_alexiss_224',
'mats/billie_eilish_224',
'mats/brie_larson_224',
'mats/cara_delevingne_224',
'mats/carolin_kebekus_224',
'mats/chelsea_clinton_224',
'mats/claire_boucher_224',
'mats/corinna_kopf_224',
'mats/florence_pugh_224',
'mats/hillary_clinton_224',
'mats/jenna_fischer_224',
'mats/kim_jisoo_320',
'mats/mica_suarez_320',
'mats/shailene_woodley_224',
'mats/shraddha_kapoor_320',
'mats/yu_jimin_352'
]
expression_restorer_models : List[ExpressionRestorerModel] = [ 'live_portrait' ]
face_debugger_items : List[FaceDebuggerItem] = [ 'bounding-box', 'face-landmark-5', 'face-landmark-5/68', 'face-landmark-68', 'face-landmark-68/5', 'face-mask', 'face-detector-score', 'face-landmarker-score', 'age', 'gender', 'race' ]

82
facefusion/processors/modules/deep_swapper.py
View File

@ -27,42 +27,56 @@ from facefusion.vision import conditional_match_frame_color, read_image, read_st
def create_model_set() -> ModelSet:
model_config =\
[
('iperov', 'emma_watson_224', 'dfl_whole_face', (224, 224)),
('iperov', 'keanu_reeves_320', 'dfl_whole_face', (320, 320)),
('iperov', 'jackie_chan_224', 'dfl_whole_face', (224, 224)),
('iperov', 'alexandra_daddario_224', 'dfl_whole_face', (224, 224)),
('iperov', 'alexei_navalny_224', 'dfl_whole_face', (224, 224)),
('iperov', 'amber_heard_224', 'dfl_whole_face', (224, 224)),
('iperov', 'dilraba_dilmurat_224', 'dfl_whole_face', (224, 224)),
('iperov', 'elon_musk_224', 'dfl_whole_face', (224, 224)),
('iperov', 'emilia_clarke_224', 'dfl_whole_face', (224, 224)),
('iperov', 'emma_watson_224', 'dfl_whole_face', (224, 224)),
('iperov', 'erin_moriarty_224', 'dfl_whole_face', (224, 224)),
('iperov', 'jackie_chan_224', 'dfl_whole_face', (224, 224)),
('iperov', 'james_carrey_224', 'dfl_whole_face', (224, 224)),
('iperov', 'jason_statham_320', 'dfl_whole_face', (320, 320)),
('iperov', 'jessica_alba_224', 'dfl_whole_face', (224, 224)),
('iperov', 'keanu_reeves_320', 'dfl_whole_face', (320, 320)),
('iperov', 'lucy_liu_224', 'dfl_whole_face', (224, 224)),
('iperov', 'margot_robbie_224', 'dfl_whole_face', (224, 224)),
('iperov', 'meghan_markle_224', 'dfl_whole_face', (224, 224)),
('iperov', 'natalie_dormer_224', 'dfl_whole_face', (224, 224)),
('iperov', 'natalie_portman_224', 'dfl_whole_face', (224, 224)),
('iperov', 'nicolas_coppola__224', 'dfl_whole_face', (224, 224)),
('iperov', 'robert_downey_224', 'dfl_whole_face', (224, 224)),
('iperov', 'rowan_atkinson_224', 'dfl_whole_face', (224, 224)),
('iperov', 'ryan_reynolds_224', 'dfl_whole_face', (224, 224)),
('iperov', 'scarlett_johansson_224', 'dfl_whole_face', (224, 224)),
('iperov', 'sylvester_stallone_224', 'dfl_whole_face', (224, 224)),
('iperov', 'taylor_swift_224', 'dfl_whole_face', (224, 224)),
('iperov', 'thomas_cruise_224', 'dfl_whole_face', (224, 224)),
('iperov', 'thomas_holland_224', 'dfl_whole_face', (224, 224)),
('iperov', 'vin_diesel_224', 'dfl_whole_face', (224, 224)),
('iperov', 'vladimir_putin_224', 'dfl_whole_face', (224, 224))
('iperov', 'alexandra_daddario_224', (224, 224)),
('iperov', 'alexei_navalny_224', (224, 224)),
('iperov', 'amber_heard_224', (224, 224)),
('iperov', 'dilraba_dilmurat_224', (224, 224)),
('iperov', 'elon_musk_224', (224, 224)),
('iperov', 'emilia_clarke_224', (224, 224)),
('iperov', 'emma_watson_224', (224, 224)),
('iperov', 'erin_moriarty_224', (224, 224)),
('iperov', 'jackie_chan_224', (224, 224)),
('iperov', 'james_carrey_224', (224, 224)),
('iperov', 'jason_statham_320', (320, 320)),
('iperov', 'jessica_alba_224', (224, 224)),
('iperov', 'keanu_reeves_320', (320, 320)),
('iperov', 'lucy_liu_224', (224, 224)),
('iperov', 'margot_robbie_224', (224, 224)),
('iperov', 'meghan_markle_224', (224, 224)),
('iperov', 'natalie_dormer_224', (224, 224)),
('iperov', 'natalie_portman_224', (224, 224)),
('iperov', 'nicolas_coppola_224', (224, 224)),
('iperov', 'robert_downey_224', (224, 224)),
('iperov', 'rowan_atkinson_224', (224, 224)),
('iperov', 'ryan_reynolds_224', (224, 224)),
('iperov', 'scarlett_johansson_224', (224, 224)),
('iperov', 'sylvester_stallone_224', (224, 224)),
('iperov', 'taylor_swift_224', (224, 224)),
('iperov', 'thomas_cruise_224', (224, 224)),
('iperov', 'thomas_holland_224', (224, 224)),
('iperov', 'vin_diesel_224', (224, 224)),
('iperov', 'vladimir_putin_224', (224, 224)),
('mats', 'alica_schmidt_320', (320, 320)),
('mats', 'ashley_alexiss_224', (224, 224)),
('mats', 'billie_eilish_224', (224, 224)),
('mats', 'brie_larson_224', (224, 224)),
('mats', 'cara_delevingne_224', (224, 224)),
('mats', 'carolin_kebekus_224', (224, 224)),
('mats', 'chelsea_clinton_224', (224, 224)),
('mats', 'claire_boucher_224', (224, 224)),
('mats', 'corinna_kopf_224', (224, 224)),
('mats', 'florence_pugh_224', (224, 224)),
('mats', 'hillary_clinton_224', (224, 224)),
('mats', 'jenna_fischer_224', (224, 224)),
('mats', 'kim_jisoo_320', (320, 320)),
('mats', 'mica_suarez_320', (320, 320)),
('mats', 'shailene_woodley_224', (224, 224)),
('mats', 'shraddha_kapoor_320', (320, 320)),
('mats', 'yu_jimin_352', (352, 352))
]
model_set : ModelSet = {}
for model_creator, model_name, model_template, model_size in model_config:
for model_creator, model_name, model_size in model_config:
model_id = '/'.join([ model_creator, model_name ])
model_set[model_id] =\
@ -83,7 +97,7 @@ def create_model_set() -> ModelSet:
'path': resolve_relative_path('../.assets/models/' + model_creator + '/' + model_name + '.dfm')
}
},
'template': model_template,
'template': 'dfl_whole_face',
'size': model_size
}

24
facefusion/processors/typing.py
View File

@ -7,9 +7,6 @@ from facefusion.typing import AppContext, AudioFrame, Face, FaceSet, VisionFrame
AgeModifierModel = Literal['styleganex_age']
DeepSwapperModel = Literal\
[
'iperov/emma_watson_224',
'iperov/keanu_reeves_320',
'iperov/jackie_chan_224',
'iperov/alexandra_daddario_224',
'iperov/alexei_navalny_224',
'iperov/amber_heard_224',
@ -28,7 +25,7 @@ DeepSwapperModel = Literal\
'iperov/meghan_markle_224',
'iperov/natalie_dormer_224',
'iperov/natalie_portman_224',
'iperov/nicolas_coppola__224',
'iperov/nicolas_coppola_224',
'iperov/robert_downey_224',
'iperov/rowan_atkinson_224',
'iperov/ryan_reynolds_224',
@ -38,7 +35,24 @@ DeepSwapperModel = Literal\
'iperov/thomas_cruise_224',
'iperov/thomas_holland_224',
'iperov/vin_diesel_224',
'iperov/vladimir_putin_224'
'iperov/vladimir_putin_224',
'mats/alica_schmidt_320',
'mats/ashley_alexiss_224',
'mats/billie_eilish_224',
'mats/brie_larson_224',
'mats/cara_delevingne_224',
'mats/carolin_kebekus_224',
'mats/chelsea_clinton_224',
'mats/claire_boucher_224',
'mats/corinna_kopf_224',
'mats/florence_pugh_224',
'mats/hillary_clinton_224',
'mats/jenna_fischer_224',
'mats/kim_jisoo_320',
'mats/mica_suarez_320',
'mats/shailene_woodley_224',
'mats/shraddha_kapoor_320',
'mats/yu_jimin_352'
]
ExpressionRestorerModel = Literal['live_portrait']
FaceDebuggerItem = Literal['bounding-box', 'face-landmark-5', 'face-landmark-5/68', 'face-landmark-68', 'face-landmark-68/5', 'face-mask', 'face-detector-score', 'face-landmarker-score', 'age', 'gender', 'race']