From c3f58b2d0f923272074d400d145ba63eaee94abb Mon Sep 17 00:00:00 2001 From: henryruhs Date: Mon, 18 Nov 2024 23:38:55 +0100 Subject: [PATCH] Add deep swapper models by Rumateus --- facefusion/processors/choices.py | 29 ++++++++++++++++++- facefusion/processors/modules/deep_swapper.py | 29 ++++++++++++++++++- facefusion/processors/typing.py | 29 ++++++++++++++++++- 3 files changed, 84 insertions(+), 3 deletions(-) diff --git a/facefusion/processors/choices.py b/facefusion/processors/choices.py index c9469f5a..5b55ae3c 100755 --- a/facefusion/processors/choices.py +++ b/facefusion/processors/choices.py @@ -117,7 +117,34 @@ deep_swapper_models : List[DeepSwapperModel] =\ 'mats/mica_suarez_320', 'mats/shailene_woodley_224', 'mats/shraddha_kapoor_320', - 'mats/yu_jimin_352' + 'mats/yu_jimin_352', + 'rumateus/alison_brie_224', + 'rumateus/amber_heard_224', + 'rumateus/angelina_jolie_224', + 'rumateus/aubrey_plaza_224', + 'rumateus/bridget_regan_224', + 'rumateus/cobie_smulders_224', + 'rumateus/deborah_woll_224', + 'rumateus/dua_lipa_224', + 'rumateus/emma_stone_224', + 'rumateus/hailee_steinfeld_224', + 'rumateus/hilary_duff_224', + 'rumateus/jessica_alba_224', + 'rumateus/jessica_biel_224', + 'rumateus/john_cena_224', + 'rumateus/kim_kardashian_224', + 'rumateus/kristen_bell_224', + 'rumateus/lucy_liu_224', + 'rumateus/margot_robbie_224', + 'rumateus/megan_fox_224', + 'rumateus/meghan_markle_224', + 'rumateus/millie_bobby_brown_224', + 'rumateus/natalie_portman_224', + 'rumateus/nicki_minaj_224', + 'rumateus/olivia_wilde_224', + 'rumateus/shay_mitchell_224', + 'rumateus/sophie_turner_224', + 'rumateus/taylor_swift_224' ] 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' ] diff --git a/facefusion/processors/modules/deep_swapper.py b/facefusion/processors/modules/deep_swapper.py index 7841c2cf..c19f8131 100755 --- a/facefusion/processors/modules/deep_swapper.py +++ b/facefusion/processors/modules/deep_swapper.py @@ -138,7 +138,34 @@ def create_model_set() -> ModelSet: ('mats', 'mica_suarez_320', (320, 320)), ('mats', 'shailene_woodley_224', (224, 224)), ('mats', 'shraddha_kapoor_320', (320, 320)), - ('mats', 'yu_jimin_352', (352, 352)) + ('mats', 'yu_jimin_352', (352, 352)), + ('rumateus', 'alison_brie_224', (224, 224)), + ('rumateus', 'amber_heard_224', (224, 224)), + ('rumateus', 'angelina_jolie_224', (224, 224)), + ('rumateus', 'aubrey_plaza_224', (224, 224)), + ('rumateus', 'bridget_regan_224', (224, 224)), + ('rumateus', 'cobie_smulders_224', (224, 224)), + ('rumateus', 'deborah_woll_224', (224, 224)), + ('rumateus', 'dua_lipa_224', (224, 224)), + ('rumateus', 'emma_stone_224', (224, 224)), + ('rumateus', 'hailee_steinfeld_224', (224, 224)), + ('rumateus', 'hilary_duff_224', (224, 224)), + ('rumateus', 'jessica_alba_224', (224, 224)), + ('rumateus', 'jessica_biel_224', (224, 224)), + ('rumateus', 'john_cena_224', (224, 224)), + ('rumateus', 'kim_kardashian_224', (224, 224)), + ('rumateus', 'kristen_bell_224', (224, 224)), + ('rumateus', 'lucy_liu_224', (224, 224)), + ('rumateus', 'margot_robbie_224', (224, 224)), + ('rumateus', 'megan_fox_224', (224, 224)), + ('rumateus', 'meghan_markle_224', (224, 224)), + ('rumateus', 'millie_bobby_brown_224', (224, 224)), + ('rumateus', 'natalie_portman_224', (224, 224)), + ('rumateus', 'nicki_minaj_224', (224, 224)), + ('rumateus', 'olivia_wilde_224', (224, 224)), + ('rumateus', 'shay_mitchell_224', (224, 224)), + ('rumateus', 'sophie_turner_224', (224, 224)), + ('rumateus', 'taylor_swift__224', (224, 224)) ] model_set : ModelSet = {} diff --git a/facefusion/processors/typing.py b/facefusion/processors/typing.py index b9eb3f31..2392fbe9 100644 --- a/facefusion/processors/typing.py +++ b/facefusion/processors/typing.py @@ -118,7 +118,34 @@ DeepSwapperModel = Literal\ 'mats/mica_suarez_320', 'mats/shailene_woodley_224', 'mats/shraddha_kapoor_320', - 'mats/yu_jimin_352' + 'mats/yu_jimin_352', + 'rumateus/alison_brie_224', + 'rumateus/amber_heard_224', + 'rumateus/angelina_jolie_224', + 'rumateus/aubrey_plaza_224', + 'rumateus/bridget_regan_224', + 'rumateus/cobie_smulders_224', + 'rumateus/deborah_woll_224', + 'rumateus/dua_lipa_224', + 'rumateus/emma_stone_224', + 'rumateus/hailee_steinfeld_224', + 'rumateus/hilary_duff_224', + 'rumateus/jessica_alba_224', + 'rumateus/jessica_biel_224', + 'rumateus/john_cena_224', + 'rumateus/kim_kardashian_224', + 'rumateus/kristen_bell_224', + 'rumateus/lucy_liu_224', + 'rumateus/margot_robbie_224', + 'rumateus/megan_fox_224', + 'rumateus/meghan_markle_224', + 'rumateus/millie_bobby_brown_224', + 'rumateus/natalie_portman_224', + 'rumateus/nicki_minaj_224', + 'rumateus/olivia_wilde_224', + 'rumateus/shay_mitchell_224', + 'rumateus/sophie_turner_224', + 'rumateus/taylor_swift_224' ] 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']