Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 01:04:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #4958 from QualiaRain/fix/control-dwpose-legacy-model-noop

Browse Source 
fix(control): remove inert DWPose (Legacy) pose-model selector
This commit is contained in:
Vladimir Mandic
2026-06-22 16:40:08 +02:00
committed by GitHub
parent 6e6ed82732 c0f5f41042
commit 0485e8708f
2 changed files with 12 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/control/processors.py
+12 -13
View File
@@ -52,7 +52,7 @@ config = {
'Shuffle': {'class': None, 'group': 'Other', 'checkpoint': False, 'params': {}},
# legacy models
'MediaPipe Face (Legacy)': {'class': None, 'group': 'Pose', 'checkpoint': False, 'params': {'max_faces': 1, 'min_confidence': 0.5}},
'DWPose (Legacy)': {'class': None, 'group': 'Pose', 'checkpoint': False, 'model': 'Tiny', 'params': {'min_confidence': 0.3}},
'DWPose (Legacy)': {'class': None, 'group': 'Pose', 'checkpoint': False, 'params': {'min_confidence': 0.3}},
'TEED (Legacy)': {'class': None, 'group': 'Edge', 'checkpoint': True, 'load_config': {'pretrained_model_or_path': 'fal/teed'}, 'params': {}},
'Anyline (Legacy)': {'class': None, 'group': 'Edge', 'checkpoint': True, 'load_config': {'pretrained_model_or_path': 'TheMistoAI/MistoLine'}, 'params': {}},
'Normal Bae (Legacy)': {'class': None, 'group': 'Normal', 'checkpoint': True, 'params': {}},
@@ -98,7 +98,7 @@ def delay_load_config():
# pose models
'OpenPose': {'class': OpenposeDetector, 'group': 'Pose', 'checkpoint': True, 'params': {'include_body': True, 'include_hand': False, 'include_face': False}},
'MediaPipe Face (Legacy)': {'class': MediapipeFaceDetector, 'group': 'Pose', 'checkpoint': False, 'params': {'max_faces': 1, 'min_confidence': 0.5}},
'DWPose (Legacy)': {'class': RtmlibPoseDetector, 'group': 'Pose', 'checkpoint': False, 'model': 'Tiny', 'params': {'min_confidence': 0.3}},
'DWPose (Legacy)': {'class': RtmlibPoseDetector, 'group': 'Pose', 'checkpoint': False, 'params': {'min_confidence': 0.3}},
'RTMW': {'class': RtmlibPoseDetector, 'group': 'Pose', 'checkpoint': False, 'params': {'min_confidence': 0.3, 'draw_body_pose': True, 'draw_hand_pose': True, 'draw_face_pose': True}},
'RTMO': {'class': RtmlibPoseDetector, 'group': 'Pose', 'checkpoint': False, 'params': {'min_confidence': 0.3}},
'ViTPose': {'class': ViTPoseDetector, 'group': 'Pose', 'checkpoint': True, 'load_config': {'pretrained_model_or_path': 'usyd-community/vitpose-plus-base'}, 'params': {'min_confidence': 0.3}},
@@ -183,17 +183,16 @@ def update_settings(*settings):
update(['MediaPipe Face (Legacy)', 'params', 'min_confidence'], settings[15])
update(['Canny', 'params', 'low_threshold'], settings[16])
update(['Canny', 'params', 'high_threshold'], settings[17])
update(['DWPose (Legacy)', 'model'], settings[18])
update(['DWPose (Legacy)', 'params', 'min_confidence'], settings[19])
update(['SegmentAnything 1.0', 'model'], settings[20])
update(['Edge', 'params', 'pf'], settings[21])
update(['Edge', 'params', 'mode'], settings[22])
update(['Zoe Depth', 'params', 'gamma_corrected'], settings[23])
update(['Marigold Depth', 'params', 'color_map'], settings[24])
update(['Marigold Depth', 'params', 'denoising_steps'], settings[25])
update(['Marigold Depth', 'params', 'ensemble_size'], settings[26])
update(['Depth Anything', 'params', 'color_map'], settings[27])
update(['Depth Pro', 'params', 'color_map'], settings[28])
update(['DWPose (Legacy)', 'params', 'min_confidence'], settings[18])
update(['SegmentAnything 1.0', 'model'], settings[19])
update(['Edge', 'params', 'pf'], settings[20])
update(['Edge', 'params', 'mode'], settings[21])
update(['Zoe Depth', 'params', 'gamma_corrected'], settings[22])
update(['Marigold Depth', 'params', 'color_map'], settings[23])
update(['Marigold Depth', 'params', 'denoising_steps'], settings[24])
update(['Marigold Depth', 'params', 'ensemble_size'], settings[25])
update(['Depth Anything', 'params', 'color_map'], settings[26])
update(['Depth Pro', 'params', 'color_map'], settings[27])
class Processor:
modules/ui_control_elements.py
-1
View File
@@ -307,7 +307,6 @@ def create_ui_elements(units, result_txt, output_gallery):
settings.append(gr.Slider(label="Low threshold", minimum=0, maximum=1000, step=1, value=100))
settings.append(gr.Slider(label="High threshold", minimum=0, maximum=1000, step=1, value=200))
with gr.Accordion('DWPose', open=True, elem_classes=['processor-settings']):
settings.append(gr.Radio(label="Pose Model", choices=['Tiny', 'Medium', 'Large'], value='Tiny'))
settings.append(gr.Slider(label="Pose confidence", minimum=0.0, maximum=1.0, step=0.01, value=0.3))
with gr.Accordion('SegmentAnything 1.0', open=True, elem_classes=['processor-settings']):
settings.append(gr.Radio(label="Segment Model", choices=['Base', 'Large'], value='Base'))