From c18d7a4d4f19f6ff865f6d3ddf3a6fec0c6787df Mon Sep 17 00:00:00 2001 From: henryruhs Date: Mon, 18 Nov 2024 23:19:37 +0100 Subject: [PATCH] Add deep swapper models by Druuzil --- facefusion/processors/choices.py | 66 +++++++++++++++++++ facefusion/processors/modules/deep_swapper.py | 66 +++++++++++++++++++ facefusion/processors/typing.py | 66 +++++++++++++++++++ 3 files changed, 198 insertions(+) diff --git a/facefusion/processors/choices.py b/facefusion/processors/choices.py index fb14bca8..c9469f5a 100755 --- a/facefusion/processors/choices.py +++ b/facefusion/processors/choices.py @@ -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', diff --git a/facefusion/processors/modules/deep_swapper.py b/facefusion/processors/modules/deep_swapper.py index 52f91634..7841c2cf 100755 --- a/facefusion/processors/modules/deep_swapper.py +++ b/facefusion/processors/modules/deep_swapper.py @@ -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)), diff --git a/facefusion/processors/typing.py b/facefusion/processors/typing.py index 2eec87cb..b9eb3f31 100644 --- a/facefusion/processors/typing.py +++ b/facefusion/processors/typing.py @@ -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',