restructure config for processors

This commit is contained in:
henryruhs
2026-08-23 23:24:55 +02:00
parent 161aec62d1
commit 038802ae2b
12 changed files with 61 additions and 39 deletions
+22
View File
@@ -75,17 +75,29 @@ workflow_strategy =
[processors]
processors =
[processors.age_modifier]
age_modifier_model =
age_modifier_direction =
[processors.background_remover]
background_remover_model =
background_remover_fill_color =
background_remover_despill_color =
[processors.deep_swapper]
deep_swapper_model =
deep_swapper_morph =
[processors.expression_restorer]
expression_restorer_model =
expression_restorer_factor =
expression_restorer_areas =
[processors.face_debugger]
face_debugger_items =
[processors.face_editor]
face_editor_model =
face_editor_eyebrow_direction =
face_editor_eye_gaze_horizontal =
@@ -101,17 +113,27 @@ face_editor_mouth_position_vertical =
face_editor_head_pitch =
face_editor_head_yaw =
face_editor_head_roll =
[processors.face_enhancer]
face_enhancer_model =
face_enhancer_blend =
face_enhancer_weight =
[processors.face_swapper]
face_swapper_model =
face_swapper_pixel_boost =
face_swapper_weight =
[processors.frame_colorizer]
frame_colorizer_model =
frame_colorizer_size =
frame_colorizer_blend =
[processors.frame_enhancer]
frame_enhancer_model =
frame_enhancer_blend =
[processors.lip_syncer]
lip_syncer_model =
lip_syncer_weight =
@@ -130,14 +130,14 @@ def register_args(program : ArgumentParser) -> None:
group_processors.add_argument(
'--age-modifier-model',
help = translator.get('help.model', __package__),
default = config.get_str_value('processors', 'age_modifier_model', 'fran'),
default = config.get_str_value('processors.age_modifier', 'age_modifier_model', 'fran'),
choices = age_modifier_choices.age_modifier_models
),
group_processors.add_argument(
'--age-modifier-direction',
help = translator.get('help.direction', __package__),
type = int,
default = config.get_int_value('processors', 'age_modifier_direction', '0'),
default = config.get_int_value('processors.age_modifier', 'age_modifier_direction', '0'),
choices = age_modifier_choices.age_modifier_direction_range,
metavar = create_int_metavar(age_modifier_choices.age_modifier_direction_range)
)
@@ -501,21 +501,21 @@ def register_args(program : ArgumentParser) -> None:
group_processors.add_argument(
'--background-remover-model',
help = translator.get('help.model', __package__),
default = config.get_str_value('processors', 'background_remover_model', 'modnet'),
default = config.get_str_value('processors.background_remover', 'background_remover_model', 'modnet'),
choices = background_remover_choices.background_remover_models
),
group_processors.add_argument(
'--background-remover-fill-color',
help = translator.get('help.fill_color', __package__),
type = partial(sanitize_int_range, int_range = background_remover_choices.background_remover_color_range),
default = config.get_int_list('processors', 'background_remover_fill_color', '0 0 0 0'),
default = config.get_int_list('processors.background_remover', 'background_remover_fill_color', '0 0 0 0'),
nargs = '+'
),
group_processors.add_argument(
'--background-remover-despill-color',
help = translator.get('help.despill_color', __package__),
type = partial(sanitize_int_range, int_range = background_remover_choices.background_remover_color_range),
default = config.get_int_list('processors', 'background_remover_despill_color', '0 0 0 0'),
default = config.get_int_list('processors.background_remover', 'background_remover_despill_color', '0 0 0 0'),
nargs = '+'
)
],
@@ -282,14 +282,14 @@ def register_args(program : ArgumentParser) -> None:
group_processors.add_argument(
'--deep-swapper-model',
help = translator.get('help.model', __package__),
default = config.get_str_value('processors', 'deep_swapper_model', 'iperov/elon_musk_224'),
default = config.get_str_value('processors.deep_swapper', 'deep_swapper_model', 'iperov/elon_musk_224'),
choices = deep_swapper_choices.deep_swapper_models
),
group_processors.add_argument(
'--deep-swapper-morph',
help = translator.get('help.morph', __package__),
type = int,
default = config.get_int_value('processors', 'deep_swapper_morph', '100'),
default = config.get_int_value('processors.deep_swapper', 'deep_swapper_morph', '100'),
choices = deep_swapper_choices.deep_swapper_morph_range,
metavar = create_int_metavar(deep_swapper_choices.deep_swapper_morph_range)
)
@@ -105,21 +105,21 @@ def register_args(program : ArgumentParser) -> None:
group_processors.add_argument(
'--expression-restorer-model',
help = translator.get('help.model', __package__),
default = config.get_str_value('processors', 'expression_restorer_model', 'live_portrait'),
default = config.get_str_value('processors.expression_restorer', 'expression_restorer_model', 'live_portrait'),
choices = expression_restorer_choices.expression_restorer_models
),
group_processors.add_argument(
'--expression-restorer-factor',
help = translator.get('help.factor', __package__),
type = int,
default = config.get_int_value('processors', 'expression_restorer_factor', '80'),
default = config.get_int_value('processors.expression_restorer', 'expression_restorer_factor', '80'),
choices = expression_restorer_choices.expression_restorer_factor_range,
metavar = create_int_metavar(expression_restorer_choices.expression_restorer_factor_range)
),
group_processors.add_argument(
'--expression-restorer-areas',
help = translator.get('help.areas', __package__).format(choices = ', '.join(expression_restorer_choices.expression_restorer_areas)),
default = config.get_str_list('processors', 'expression_restorer_areas', ' '.join(expression_restorer_choices.expression_restorer_areas)),
default = config.get_str_list('processors.expression_restorer', 'expression_restorer_areas', ' '.join(expression_restorer_choices.expression_restorer_areas)),
choices = expression_restorer_choices.expression_restorer_areas,
nargs = '+',
metavar = 'EXPRESSION_RESTORER_AREAS'
@@ -38,7 +38,7 @@ def register_args(program : ArgumentParser) -> None:
group_processors.add_argument(
'--face-debugger-items',
help = translator.get('help.items', __package__).format(choices = ', '.join(face_debugger_choices.face_debugger_items)),
default = config.get_str_list('processors', 'face_debugger_items', 'face-landmark-5/68 face-mask'),
default = config.get_str_list('processors.face_debugger', 'face_debugger_items', 'face-landmark-5/68 face-mask'),
choices = face_debugger_choices.face_debugger_items,
nargs = '+',
metavar = 'FACE_DEBUGGER_ITEMS'
@@ -135,14 +135,14 @@ def register_args(program : ArgumentParser) -> None:
group_processors.add_argument(
'--face-editor-model',
help = translator.get('help.model', __package__),
default = config.get_str_value('processors', 'face_editor_model', 'live_portrait'),
default = config.get_str_value('processors.face_editor', 'face_editor_model', 'live_portrait'),
choices = face_editor_choices.face_editor_models
),
group_processors.add_argument(
'--face-editor-eyebrow-direction',
help = translator.get('help.eyebrow_direction', __package__),
type = float,
default = config.get_float_value('processors', 'face_editor_eyebrow_direction', '0'),
default = config.get_float_value('processors.face_editor', 'face_editor_eyebrow_direction', '0'),
choices = face_editor_choices.face_editor_eyebrow_direction_range,
metavar = create_float_metavar(face_editor_choices.face_editor_eyebrow_direction_range)
),
@@ -150,7 +150,7 @@ def register_args(program : ArgumentParser) -> None:
'--face-editor-eye-gaze-horizontal',
help = translator.get('help.eye_gaze_horizontal', __package__),
type = float,
default = config.get_float_value('processors', 'face_editor_eye_gaze_horizontal', '0'),
default = config.get_float_value('processors.face_editor', 'face_editor_eye_gaze_horizontal', '0'),
choices = face_editor_choices.face_editor_eye_gaze_horizontal_range,
metavar = create_float_metavar(face_editor_choices.face_editor_eye_gaze_horizontal_range)
),
@@ -158,7 +158,7 @@ def register_args(program : ArgumentParser) -> None:
'--face-editor-eye-gaze-vertical',
help = translator.get('help.eye_gaze_vertical', __package__),
type = float,
default = config.get_float_value('processors', 'face_editor_eye_gaze_vertical', '0'),
default = config.get_float_value('processors.face_editor', 'face_editor_eye_gaze_vertical', '0'),
choices = face_editor_choices.face_editor_eye_gaze_vertical_range,
metavar = create_float_metavar(face_editor_choices.face_editor_eye_gaze_vertical_range)
),
@@ -166,7 +166,7 @@ def register_args(program : ArgumentParser) -> None:
'--face-editor-eye-open-ratio',
help = translator.get('help.eye_open_ratio', __package__),
type = float,
default = config.get_float_value('processors', 'face_editor_eye_open_ratio', '0'),
default = config.get_float_value('processors.face_editor', 'face_editor_eye_open_ratio', '0'),
choices = face_editor_choices.face_editor_eye_open_ratio_range,
metavar = create_float_metavar(face_editor_choices.face_editor_eye_open_ratio_range)
),
@@ -174,7 +174,7 @@ def register_args(program : ArgumentParser) -> None:
'--face-editor-lip-open-ratio',
help = translator.get('help.lip_open_ratio', __package__),
type = float,
default = config.get_float_value('processors', 'face_editor_lip_open_ratio', '0'),
default = config.get_float_value('processors.face_editor', 'face_editor_lip_open_ratio', '0'),
choices = face_editor_choices.face_editor_lip_open_ratio_range,
metavar = create_float_metavar(face_editor_choices.face_editor_lip_open_ratio_range)
),
@@ -182,7 +182,7 @@ def register_args(program : ArgumentParser) -> None:
'--face-editor-mouth-grim',
help = translator.get('help.mouth_grim', __package__),
type = float,
default = config.get_float_value('processors', 'face_editor_mouth_grim', '0'),
default = config.get_float_value('processors.face_editor', 'face_editor_mouth_grim', '0'),
choices = face_editor_choices.face_editor_mouth_grim_range,
metavar = create_float_metavar(face_editor_choices.face_editor_mouth_grim_range)
),
@@ -190,7 +190,7 @@ def register_args(program : ArgumentParser) -> None:
'--face-editor-mouth-pout',
help = translator.get('help.mouth_pout', __package__),
type = float,
default = config.get_float_value('processors', 'face_editor_mouth_pout', '0'),
default = config.get_float_value('processors.face_editor', 'face_editor_mouth_pout', '0'),
choices = face_editor_choices.face_editor_mouth_pout_range,
metavar = create_float_metavar(face_editor_choices.face_editor_mouth_pout_range)
),
@@ -198,7 +198,7 @@ def register_args(program : ArgumentParser) -> None:
'--face-editor-mouth-purse',
help = translator.get('help.mouth_purse', __package__),
type = float,
default = config.get_float_value('processors', 'face_editor_mouth_purse', '0'),
default = config.get_float_value('processors.face_editor', 'face_editor_mouth_purse', '0'),
choices = face_editor_choices.face_editor_mouth_purse_range,
metavar = create_float_metavar(face_editor_choices.face_editor_mouth_purse_range)
),
@@ -206,7 +206,7 @@ def register_args(program : ArgumentParser) -> None:
'--face-editor-mouth-smile',
help = translator.get('help.mouth_smile', __package__),
type = float,
default = config.get_float_value('processors', 'face_editor_mouth_smile', '0'),
default = config.get_float_value('processors.face_editor', 'face_editor_mouth_smile', '0'),
choices = face_editor_choices.face_editor_mouth_smile_range,
metavar = create_float_metavar(face_editor_choices.face_editor_mouth_smile_range)
),
@@ -214,7 +214,7 @@ def register_args(program : ArgumentParser) -> None:
'--face-editor-mouth-position-horizontal',
help = translator.get('help.mouth_position_horizontal', __package__),
type = float,
default = config.get_float_value('processors', 'face_editor_mouth_position_horizontal', '0'),
default = config.get_float_value('processors.face_editor', 'face_editor_mouth_position_horizontal', '0'),
choices = face_editor_choices.face_editor_mouth_position_horizontal_range,
metavar = create_float_metavar(face_editor_choices.face_editor_mouth_position_horizontal_range)
),
@@ -222,7 +222,7 @@ def register_args(program : ArgumentParser) -> None:
'--face-editor-mouth-position-vertical',
help = translator.get('help.mouth_position_vertical', __package__),
type = float,
default = config.get_float_value('processors', 'face_editor_mouth_position_vertical', '0'),
default = config.get_float_value('processors.face_editor', 'face_editor_mouth_position_vertical', '0'),
choices = face_editor_choices.face_editor_mouth_position_vertical_range,
metavar = create_float_metavar(face_editor_choices.face_editor_mouth_position_vertical_range)
),
@@ -230,7 +230,7 @@ def register_args(program : ArgumentParser) -> None:
'--face-editor-head-pitch',
help = translator.get('help.head_pitch', __package__),
type = float,
default = config.get_float_value('processors', 'face_editor_head_pitch', '0'),
default = config.get_float_value('processors.face_editor', 'face_editor_head_pitch', '0'),
choices = face_editor_choices.face_editor_head_pitch_range,
metavar = create_float_metavar(face_editor_choices.face_editor_head_pitch_range)
),
@@ -238,7 +238,7 @@ def register_args(program : ArgumentParser) -> None:
'--face-editor-head-yaw',
help = translator.get('help.head_yaw', __package__),
type = float,
default = config.get_float_value('processors', 'face_editor_head_yaw', '0'),
default = config.get_float_value('processors.face_editor', 'face_editor_head_yaw', '0'),
choices = face_editor_choices.face_editor_head_yaw_range,
metavar = create_float_metavar(face_editor_choices.face_editor_head_yaw_range)
),
@@ -246,7 +246,7 @@ def register_args(program : ArgumentParser) -> None:
'--face-editor-head-roll',
help = translator.get('help.head_roll', __package__),
type = float,
default = config.get_float_value('processors', 'face_editor_head_roll', '0'),
default = config.get_float_value('processors.face_editor', 'face_editor_head_roll', '0'),
choices = face_editor_choices.face_editor_head_roll_range,
metavar = create_float_metavar(face_editor_choices.face_editor_head_roll_range)
)
@@ -299,14 +299,14 @@ def register_args(program : ArgumentParser) -> None:
group_processors.add_argument(
'--face-enhancer-model',
help = translator.get('help.model', __package__),
default = config.get_str_value('processors', 'face_enhancer_model', 'gfpgan_1.4'),
default = config.get_str_value('processors.face_enhancer', 'face_enhancer_model', 'gfpgan_1.4'),
choices = face_enhancer_choices.face_enhancer_models
),
group_processors.add_argument(
'--face-enhancer-blend',
help = translator.get('help.blend', __package__),
type = int,
default = config.get_int_value('processors', 'face_enhancer_blend', '80'),
default = config.get_int_value('processors.face_enhancer', 'face_enhancer_blend', '80'),
choices = face_enhancer_choices.face_enhancer_blend_range,
metavar = create_int_metavar(face_enhancer_choices.face_enhancer_blend_range)
),
@@ -314,7 +314,7 @@ def register_args(program : ArgumentParser) -> None:
'--face-enhancer-weight',
help = translator.get('help.weight', __package__),
type = float,
default = config.get_float_value('processors', 'face_enhancer_weight', '0.5'),
default = config.get_float_value('processors.face_enhancer', 'face_enhancer_weight', '0.5'),
choices = face_enhancer_choices.face_enhancer_weight_range,
metavar = create_float_metavar(face_enhancer_choices.face_enhancer_weight_range)
)
@@ -532,7 +532,7 @@ def register_args(program : ArgumentParser) -> None:
group_processors.add_argument(
'--face-swapper-model',
help = translator.get('help.model', __package__),
default = config.get_str_value('processors', 'face_swapper_model', 'hyperswap_1a_256'),
default = config.get_str_value('processors.face_swapper', 'face_swapper_model', 'hyperswap_1a_256'),
choices = face_swapper_choices.face_swapper_models
)
],
@@ -546,14 +546,14 @@ def register_args(program : ArgumentParser) -> None:
group_processors.add_argument(
'--face-swapper-pixel-boost',
help = translator.get('help.pixel_boost', __package__),
default = config.get_str_value('processors', 'face_swapper_pixel_boost', get_first(face_swapper_pixel_boost_choices)),
default = config.get_str_value('processors.face_swapper', 'face_swapper_pixel_boost', get_first(face_swapper_pixel_boost_choices)),
choices = face_swapper_pixel_boost_choices
),
group_processors.add_argument(
'--face-swapper-weight',
help = translator.get('help.weight', __package__),
type = float,
default = config.get_float_value('processors', 'face_swapper_weight', '0.5'),
default = config.get_float_value('processors.face_swapper', 'face_swapper_weight', '0.5'),
choices = face_swapper_choices.face_swapper_weight_range
)
],
@@ -192,21 +192,21 @@ def register_args(program : ArgumentParser) -> None:
group_processors.add_argument(
'--frame-colorizer-model',
help = translator.get('help.model', __package__),
default = config.get_str_value('processors', 'frame_colorizer_model', 'ddcolor'),
default = config.get_str_value('processors.frame_colorizer', 'frame_colorizer_model', 'ddcolor'),
choices = frame_colorizer_choices.frame_colorizer_models
),
group_processors.add_argument(
'--frame-colorizer-size',
help = translator.get('help.size', __package__),
type = str,
default = config.get_str_value('processors', 'frame_colorizer_size', '256x256'),
default = config.get_str_value('processors.frame_colorizer', 'frame_colorizer_size', '256x256'),
choices = frame_colorizer_choices.frame_colorizer_sizes
),
group_processors.add_argument(
'--frame-colorizer-blend',
help = translator.get('help.blend', __package__),
type = int,
default = config.get_int_value('processors', 'frame_colorizer_blend', '100'),
default = config.get_int_value('processors.frame_colorizer', 'frame_colorizer_blend', '100'),
choices = frame_colorizer_choices.frame_colorizer_blend_range,
metavar = create_int_metavar(frame_colorizer_choices.frame_colorizer_blend_range)
)
@@ -586,14 +586,14 @@ def register_args(program : ArgumentParser) -> None:
group_processors.add_argument(
'--frame-enhancer-model',
help = translator.get('help.model', __package__),
default = config.get_str_value('processors', 'frame_enhancer_model', 'span_kendata_x4'),
default = config.get_str_value('processors.frame_enhancer', 'frame_enhancer_model', 'span_kendata_x4'),
choices = frame_enhancer_choices.frame_enhancer_models
),
group_processors.add_argument(
'--frame-enhancer-blend',
help = translator.get('help.blend', __package__),
type = int,
default = config.get_int_value('processors', 'frame_enhancer_blend', '80'),
default = config.get_int_value('processors.frame_enhancer', 'frame_enhancer_blend', '80'),
choices = frame_enhancer_choices.frame_enhancer_blend_range,
metavar = create_int_metavar(frame_enhancer_choices.frame_enhancer_blend_range)
)
@@ -139,14 +139,14 @@ def register_args(program : ArgumentParser) -> None:
group_processors.add_argument(
'--lip-syncer-model',
help = translator.get('help.model', __package__),
default = config.get_str_value('processors', 'lip_syncer_model', 'wav2lip_gan_96'),
default = config.get_str_value('processors.lip_syncer', 'lip_syncer_model', 'wav2lip_gan_96'),
choices = lip_syncer_choices.lip_syncer_models
),
group_processors.add_argument(
'--lip-syncer-weight',
help = translator.get('help.weight', __package__),
type = float,
default = config.get_float_value('processors', 'lip_syncer_weight', '0.5'),
default = config.get_float_value('processors.lip_syncer', 'lip_syncer_weight', '0.5'),
choices = lip_syncer_choices.lip_syncer_weight_range,
metavar = create_float_metavar(lip_syncer_choices.lip_syncer_weight_range)
)