8bf9170577
* Mark as NEXT * Reduce caching to avoid RAM explosion * Reduce caching to avoid RAM explosion * Update dependencies * add face-detector-pad-factor * update facefusion.ini * fix test * change pad to margin * fix order * add prepare margin * use 50% max margin * Minor fixes part2 * Minor fixes part3 * Minor fixes part4 * Minor fixes part1 * Downgrade onnxruntime as of BiRefNet broken on CPU add test update update facefusion.ini add birefnet * rename models add more models * Fix versions * Add .claude to gitignore * add normalize color add 4 channel add colors * worflows * cleanup * cleanup * cleanup * cleanup * add more models (#961) * Fix naming * changes * Fix style and mock Gradio * Fix style and mock Gradio * Fix style and mock Gradio * apply clamp * remove clamp * Add normalizer test * Introduce sanitizer for the rescue (#963) * Introduce sanitizer for the rescue * Introduce sanitizer for the rescue * Introduce sanitizer for the rescue * prepare ffmpeg for alpha support * Some cleanup * Some cleanup * Fix CI * List as TypeAlias is not allowed (#967) * List as TypeAlias is not allowed * List as TypeAlias is not allowed * List as TypeAlias is not allowed * List as TypeAlias is not allowed * Add mpeg and mxf support (#968) * Add mpeg support * Add mxf support * Adjust fix_xxx_encoder for the new formats * Extend output pattern for batch-run (#969) * Extend output pattern for batch-run * Add {target_extension} to allowed mixed files * Catch invalid output pattern keys * alpha support * cleanup * cleanup * add ProcessorOutputs type * fix preview and streamer, support alpha for background_remover * Refactor/open close processors (#972) * Introduce open/close processors * Add locales for translator * Introduce __autoload__ for translator * More cleanup * Fix import issues * Resolve the scope situation for locals * Fix installer by not using translator * Fixes after merge * Fixes after merge * Fix translator keys in ui * Use LOCALS in installer * Update and partial fix DirectML * Use latest onnxruntime * Fix performance * Fix lint issues * fix mask * fix lint * fix lint * Remove default from translator.get() * remove 'framerate=' * fix test * Rename and reorder models * Align naming * add alpha preview * fix frame-by-frame * Add alpha effect via css * preview support alpha channel * fix preview modes * Use official assets repositories * Add support for u2net_cloth * fix naming * Add more models * Add vendor, license and year direct to the models * Add vendor, license and year direct to the models * Update dependencies, Minor CSS adjustment * Ready for 3.5.0 * Fix naming * Update about messages * Fix return * Use groups to show/hide * Update preview * Conditional merge mask * Conditional merge mask * Fix import order --------- Co-authored-by: harisreedhar <h4harisreedhar.s.s@gmail.com> Co-authored-by: Harisreedhar <46858047+harisreedhar@users.noreply.github.com>
177 lines
5.2 KiB
Python
177 lines
5.2 KiB
Python
from typing import List, Sequence
|
|
|
|
from facefusion.common_helper import create_int_range
|
|
from facefusion.filesystem import get_file_name, resolve_file_paths, resolve_relative_path
|
|
from facefusion.processors.modules.deep_swapper.types import DeepSwapperModel
|
|
|
|
deep_swapper_models : List[DeepSwapperModel] =\
|
|
[
|
|
'druuzil/adam_levine_320',
|
|
'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/angelina_jolie_384',
|
|
'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/brie_larson_384',
|
|
'druuzil/bruce_campbell_384',
|
|
'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_carrey_384',
|
|
'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/luke_evans_384',
|
|
'druuzil/mads_mikkelsen_384',
|
|
'druuzil/mary_winstead_320',
|
|
'druuzil/margaret_qualley_384',
|
|
'druuzil/melina_juergens_320',
|
|
'druuzil/michael_fassbender_320',
|
|
'druuzil/michael_fox_320',
|
|
'druuzil/millie_bobby_brown_320',
|
|
'druuzil/morgan_freeman_320',
|
|
'druuzil/patrick_stewart_224',
|
|
'druuzil/rachel_weisz_384',
|
|
'druuzil/rebecca_ferguson_320',
|
|
'druuzil/scarlett_johansson_320',
|
|
'druuzil/shannen_doherty_384',
|
|
'druuzil/seth_macfarlane_384',
|
|
'druuzil/thomas_cruise_320',
|
|
'druuzil/thomas_hanks_384',
|
|
'druuzil/william_murray_384',
|
|
'druuzil/zoe_saldana_384',
|
|
'edel/emma_roberts_224',
|
|
'edel/ivanka_trump_224',
|
|
'edel/lize_dzjabrailova_224',
|
|
'edel/sidney_sweeney_224',
|
|
'edel/winona_ryder_224',
|
|
'iperov/alexandra_daddario_224',
|
|
'iperov/alexei_navalny_224',
|
|
'iperov/amber_heard_224',
|
|
'iperov/dilraba_dilmurat_224',
|
|
'iperov/elon_musk_224',
|
|
'iperov/emilia_clarke_224',
|
|
'iperov/emma_watson_224',
|
|
'iperov/erin_moriarty_224',
|
|
'iperov/jackie_chan_224',
|
|
'iperov/james_carrey_224',
|
|
'iperov/jason_statham_320',
|
|
'iperov/keanu_reeves_320',
|
|
'iperov/margot_robbie_224',
|
|
'iperov/natalie_dormer_224',
|
|
'iperov/nicolas_coppola_224',
|
|
'iperov/robert_downey_224',
|
|
'iperov/rowan_atkinson_224',
|
|
'iperov/ryan_reynolds_224',
|
|
'iperov/scarlett_johansson_224',
|
|
'iperov/sylvester_stallone_224',
|
|
'iperov/thomas_cruise_224',
|
|
'iperov/thomas_holland_224',
|
|
'iperov/vin_diesel_224',
|
|
'iperov/vladimir_putin_224',
|
|
'jen/angelica_trae_288',
|
|
'jen/ella_freya_224',
|
|
'jen/emma_myers_320',
|
|
'jen/evie_pickerill_224',
|
|
'jen/kang_hyewon_320',
|
|
'jen/maddie_mead_224',
|
|
'jen/nicole_turnbull_288',
|
|
'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',
|
|
'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'
|
|
]
|
|
|
|
custom_model_file_paths = resolve_file_paths(resolve_relative_path('../.assets/models/custom'))
|
|
|
|
if custom_model_file_paths:
|
|
|
|
for model_file_path in custom_model_file_paths:
|
|
model_id = '/'.join([ 'custom', get_file_name(model_file_path) ])
|
|
deep_swapper_models.append(model_id)
|
|
|
|
deep_swapper_morph_range : Sequence[int] = create_int_range(0, 100, 1)
|