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

Add deep swapper models by Druuzil

Browse Source
This commit is contained in:
henryruhs 2024-11-18 23:19:37 +01:00
parent b6e895fcf6
commit c18d7a4d4f
3 changed files with 198 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

66
facefusion/processors/choices.py
View File

@ -6,6 +6,72 @@ from facefusion.processors.typing import AgeModifierModel, DeepSwapperModel, Exp
age_modifier_models : List[AgeModifierModel] = [ 'styleganex_age' ]
deep_swapper_models : List[DeepSwapperModel] =\
[
'druuzil/adrianne_palicki_384',
'druuzil/agnetha_falskog_224',
'druuzil/alan_ritchson_320',
'druuzil/alicia_vikander_320',
'druuzil/amber_midthunder_320',
'druuzil/andras_arato_384',
'druuzil/andrew_tate_320',
'druuzil/anne_hathaway_320',
'druuzil/anya_chalotra_320',
'druuzil/arnold_schwarzenegger_320',
'druuzil/benjamin_affleck_320',
'druuzil/benjamin_stiller_384',
'druuzil/bradley_pitt_224',
'druuzil/bryan_cranston_320',
'druuzil/catherine_blanchett_352',
'druuzil/christian_bale_320',
'druuzil/christopher_hemsworth_320',
'druuzil/christoph_waltz_384',
'druuzil/cillian_murphy_320',
'druuzil/cobie_smulders_256',
'druuzil/dwayne_johnson_384',
'druuzil/edward_norton_320',
'druuzil/elisabeth_shue_320',
'druuzil/elizabeth_olsen_384',
'druuzil/elon_musk_320',
'druuzil/emily_blunt_320',
'druuzil/emma_stone_384',
'druuzil/emma_watson_320',
'druuzil/erin_moriarty_384',
'druuzil/eva_green_320',
'druuzil/ewan_mcgregor_320',
'druuzil/florence_pugh_320',
'druuzil/freya_allan_320',
'druuzil/gary_cole_224',
'druuzil/gigi_hadid_224',
'druuzil/harrison_ford_384',
'druuzil/hayden_christensen_320',
'druuzil/heath_ledger_320',
'druuzil/henry_cavill_448',
'druuzil/hugh_jackman_384',
'druuzil/idris_elba_320',
'druuzil/jack_nicholson_320',
'druuzil/james_mcavoy_320',
'druuzil/james_varney_320',
'druuzil/jason_momoa_320',
'druuzil/jason_statham_320',
'druuzil/jennifer_connelly_384',
'druuzil/jimmy_donaldson_320',
'druuzil/jordan_peterson_384',
'druuzil/karl_urban_224',
'druuzil/kate_beckinsale_384',
'druuzil/laurence_fishburne_384',
'druuzil/lili_reinhart_320',
'druuzil/mads_mikkelsen_384',
'druuzil/mary_winstead_320',
'druuzil/melina_juergens_320',
'druuzil/michael_fassbender_320',
'druuzil/michael_fox_320',
'druuzil/millie_bobby_brown_320',
'druuzil/morgan_freeman_320',
'druuzil/patrick_stewart_320',
'druuzil/rebecca_ferguson_320',
'druuzil/scarlett_johansson_320',
'druuzil/seth_macfarlane_384',
'druuzil/thomas_cruise_320',
'druuzil/thomas_hanks_384',
'iperov/alexandra_daddario_224',
'iperov/alexei_navalny_224',
'iperov/amber_heard_224',

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

@ -27,6 +27,72 @@ from facefusion.vision import conditional_match_frame_color, read_image, read_st
def create_model_set() -> ModelSet:
model_config =\
[
('druuzil', 'adrianne_palicki_384', (384, 384)),
('druuzil', 'agnetha_falskog_224', (224, 224)),
('druuzil', 'alan_ritchson_320', (320, 320)),
('druuzil', 'alicia_vikander_320', (320, 320)),
('druuzil', 'amber_midthunder_320', (320, 320)),
('druuzil', 'andras_arato_384', (384, 384)),
('druuzil', 'andrew_tate_320', (320, 320)),
('druuzil', 'anne_hathaway_320', (320, 320)),
('druuzil', 'anya_chalotra_320', (320, 320)),
('druuzil', 'arnold_schwarzenegger_320', (320, 320)),
('druuzil', 'benjamin_affleck_320', (320, 320)),
('druuzil', 'benjamin_stiller_384', (384, 384)),
('druuzil', 'bradley_pitt_224', (224, 224)),
('druuzil', 'bryan_cranston_320', (320, 320)),
('druuzil', 'catherine_blanchett_352', (352, 352)),
('druuzil', 'christian_bale_320', (320, 320)),
('druuzil', 'christopher_hemsworth_320', (320, 320)),
('druuzil', 'christoph_waltz_384', (384, 384)),
('druuzil', 'cillian_murphy_320', (320, 320)),
('druuzil', 'cobie_smulders_256', (256, 256)),
('druuzil', 'dwayne_johnson_384', (384, 384)),
('druuzil', 'edward_norton_320', (320, 320)),
('druuzil', 'elisabeth_shue_320', (320, 320)),
('druuzil', 'elizabeth_olsen_384', (384, 384)),
('druuzil', 'elon_musk_320', (320, 320)),
('druuzil', 'emily_blunt_320', (320, 320)),
('druuzil', 'emma_stone_384', (384, 384)),
('druuzil', 'emma_watson_320', (320, 320)),
('druuzil', 'erin_moriarty_384', (384, 384)),
('druuzil', 'eva_green_320', (320, 320)),
('druuzil', 'ewan_mcgregor_320', (320, 320)),
('druuzil', 'florence_pugh_320', (320, 320)),
('druuzil', 'freya_allan_320', (320, 320)),
('druuzil', 'gary_cole_224', (224, 224)),
('druuzil', 'gigi_hadid_224', (224, 224)),
('druuzil', 'harrison_ford_384', (384, 384)),
('druuzil', 'hayden_christensen_320', (320, 320)),
('druuzil', 'heath_ledger_320', (320, 320)),
('druuzil', 'henry_cavill_448', (448, 448)),
('druuzil', 'hugh_jackman_384', (384, 384)),
('druuzil', 'idris_elba_320', (320, 320)),
('druuzil', 'jack_nicholson_320', (320, 320)),
('druuzil', 'james_mcavoy_320', (320, 320)),
('druuzil', 'james_varney_320', (320, 320)),
('druuzil', 'jason_momoa_320', (320, 320)),
('druuzil', 'jason_statham_320', (320, 320)),
('druuzil', 'jennifer_connelly_384', (384, 384)),
('druuzil', 'jimmy_donaldson_320', (320, 320)),
('druuzil', 'jordan_peterson_384', (384, 384)),
('druuzil', 'karl_urban_224', (224, 224)),
('druuzil', 'kate_beckinsale_384', (384, 384)),
('druuzil', 'laurence_fishburne_384', (384, 384)),
('druuzil', 'lili_reinhart_320', (320, 320)),
('druuzil', 'mads_mikkelsen_384', (384, 384)),
('druuzil', 'mary_winstead_320', (320, 320)),
('druuzil', 'melina_juergens_320', (320, 320)),
('druuzil', 'michael_fassbender_320', (320, 320)),
('druuzil', 'michael_fox_320', (320, 320)),
('druuzil', 'millie_bobby_brown_320', (320, 320)),
('druuzil', 'morgan_freeman_320', (320, 320)),
('druuzil', 'patrick_stewart_320', (320, 320)),
('druuzil', 'rebecca_ferguson_320', (320, 320)),
('druuzil', 'scarlett_johansson_320', (320, 320)),
('druuzil', 'seth_macfarlane_384', (384, 384)),
('druuzil', 'thomas_cruise_320', (320, 320)),
('druuzil', 'thomas_hanks_384', (384, 384)),
('iperov', 'alexandra_daddario_224', (224, 224)),
('iperov', 'alexei_navalny_224', (224, 224)),
('iperov', 'amber_heard_224', (224, 224)),

66
facefusion/processors/typing.py
View File

@ -7,6 +7,72 @@ from facefusion.typing import AppContext, AudioFrame, Face, FaceSet, VisionFrame
AgeModifierModel = Literal['styleganex_age']
DeepSwapperModel = Literal\
[
'druuzil/adrianne_palicki_384',
'druuzil/agnetha_falskog_224',
'druuzil/alan_ritchson_320',
'druuzil/alicia_vikander_320',
'druuzil/amber_midthunder_320',
'druuzil/andras_arato_384',
'druuzil/andrew_tate_320',
'druuzil/anne_hathaway_320',
'druuzil/anya_chalotra_320',
'druuzil/arnold_schwarzenegger_320',
'druuzil/benjamin_affleck_320',
'druuzil/benjamin_stiller_384',
'druuzil/bradley_pitt_224',
'druuzil/bryan_cranston_320',
'druuzil/catherine_blanchett_352',
'druuzil/christian_bale_320',
'druuzil/christopher_hemsworth_320',
'druuzil/christoph_waltz_384',
'druuzil/cillian_murphy_320',
'druuzil/cobie_smulders_256',
'druuzil/dwayne_johnson_384',
'druuzil/edward_norton_320',
'druuzil/elisabeth_shue_320',
'druuzil/elizabeth_olsen_384',
'druuzil/elon_musk_320',
'druuzil/emily_blunt_320',
'druuzil/emma_stone_384',
'druuzil/emma_watson_320',
'druuzil/erin_moriarty_384',
'druuzil/eva_green_320',
'druuzil/ewan_mcgregor_320',
'druuzil/florence_pugh_320',
'druuzil/freya_allan_320',
'druuzil/gary_cole_224',
'druuzil/gigi_hadid_224',
'druuzil/harrison_ford_384',
'druuzil/hayden_christensen_320',
'druuzil/heath_ledger_320',
'druuzil/henry_cavill_448',
'druuzil/hugh_jackman_384',
'druuzil/idris_elba_320',
'druuzil/jack_nicholson_320',
'druuzil/james_mcavoy_320',
'druuzil/james_varney_320',
'druuzil/jason_momoa_320',
'druuzil/jason_statham_320',
'druuzil/jennifer_connelly_384',
'druuzil/jimmy_donaldson_320',
'druuzil/jordan_peterson_384',
'druuzil/karl_urban_224',
'druuzil/kate_beckinsale_384',
'druuzil/laurence_fishburne_384',
'druuzil/lili_reinhart_320',
'druuzil/mads_mikkelsen_384',
'druuzil/mary_winstead_320',
'druuzil/melina_juergens_320',
'druuzil/michael_fassbender_320',
'druuzil/michael_fox_320',
'druuzil/millie_bobby_brown_320',
'druuzil/morgan_freeman_320',
'druuzil/patrick_stewart_320',
'druuzil/rebecca_ferguson_320',
'druuzil/scarlett_johansson_320',
'druuzil/seth_macfarlane_384',
'druuzil/thomas_cruise_320',
'druuzil/thomas_hanks_384',
'iperov/alexandra_daddario_224',
'iperov/alexei_navalny_224',
'iperov/amber_heard_224',