Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

New process for poseEstimation using multiple processes #156

Open
wants to merge 14 commits into
base: main
Choose a base branch
from
11 changes: 10 additions & 1 deletion Pose2Sim/Demo_Batch/Config.toml
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,16 @@ exclude_from_batch = [] # List of trials to be excluded from batch analysis, ['<


[pose]
vid_img_extension = 'mp4' # any video or image extension
rotation = 0 # Degree of rotation applied to the image, allowed in multiples of 90. Use [90, 90, ...] for different angles per source.
vid_img_extension = 'mp4' # File extension for videos or images, or 'webcam' for webcam input.

webcam_ids = 0 # Webcam ID to use. Default is 0. For multiple webcams, use a list like [0, 1, ...].
capture_mode = 'continuous' # Only useful for webcams. 'Continuous' captures frames as soon as buffer space is available; 'alternating' waits until the buffer is free.
webcam_recording = false # If True, saves webcam streams while analyzing them to prevent frame loss due to processing delays.

combined_frames = true # Accelerates processing by batch processing frames. (Beta feature)
multi_workers = false # Enables multiple worker processes for parallel frame analysis. (Not recommended)
max_simultaneous_sources = 9 # Limits the number of sources processed simultaneously. Recommended values: 4, 9, 16, ... (perfect square).

pose_model = 'Body_with_feet' #With RTMLib:
# - Body_with_feet (default HALPE_26 model),
Expand Down
11 changes: 10 additions & 1 deletion Pose2Sim/Demo_MultiPerson/Config.toml
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,16 @@ exclude_from_batch = [] # List of trials to be excluded from batch analysis, ['<


[pose]
vid_img_extension = 'mp4' # any video or image extension
rotation = 0 # Degree of rotation applied to the image, allowed in multiples of 90. Use [90, 90, ...] for different angles per source.
vid_img_extension = 'mp4' # File extension for videos or images, or 'webcam' for webcam input.

webcam_ids = 0 # Webcam ID to use. Default is 0. For multiple webcams, use a list like [0, 1, ...].
capture_mode = 'continuous' # Only useful for webcams. 'Continuous' captures frames as soon as buffer space is available; 'alternating' waits until the buffer is free.
webcam_recording = false # If True, saves webcam streams while analyzing them to prevent frame loss due to processing delays.

combined_frames = true # Accelerates processing by batch processing frames. (Beta feature)
multi_workers = false # Enables multiple worker processes for parallel frame analysis. (Not recommended)
max_simultaneous_sources = 9 # Limits the number of sources processed simultaneously. Recommended values: 4, 9, 16, ... (perfect square).

pose_model = 'Body_with_feet' #With RTMLib:
# - Body_with_feet (default HALPE_26 model),
Expand Down
15 changes: 12 additions & 3 deletions Pose2Sim/Demo_SinglePerson/Config.toml
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,16 @@ exclude_from_batch = [] # List of trials to be excluded from batch analysis, ['<


[pose]
vid_img_extension = 'mp4' # any video or image extension
rotation = 0 # Degree of rotation applied to the image, allowed in multiples of 90. Use [90, 90, ...] for different angles per source.
vid_img_extension = 'mp4' # File extension for videos or images, or 'webcam' for webcam input.

webcam_ids = 0 # Webcam ID to use. Default is 0. For multiple webcams, use a list like [0, 1, ...].
capture_mode = 'continuous' # Only useful for webcams. 'Continuous' captures frames as soon as buffer space is available; 'alternating' waits until the buffer is free.
webcam_recording = false # If True, saves webcam streams while analyzing them to prevent frame loss due to processing delays.

combined_frames = true # Accelerates processing by batch processing frames. (Beta feature)
multi_workers = false # Enables multiple worker processes for parallel frame analysis. (Not recommended)
max_simultaneous_sources = 9 # Limits the number of sources processed simultaneously. Recommended values: 4, 9, 16, ... (perfect square).

pose_model = 'Body_with_feet' #With RTMLib:
# - Body_with_feet (default HALPE_26 model),
Expand Down Expand Up @@ -76,7 +85,7 @@ mode = 'balanced' # 'lightweight', 'balanced', 'performance',
# 'pose_input_size':[256,256]}"""


det_frequency = 4 # Run person detection only every N frames, and inbetween track previously detected bounding boxes (pose estimation is still run on all frames).
det_frequency = 1 # Run person detection only every N frames, and inbetween track previously detected bounding boxes (pose estimation is still run on all frames).
# Equal to or greater than 1, can be as high as you want in simple uncrowded cases. Much faster, but might be less accurate.
device = 'auto' # 'auto', 'CPU', 'CUDA', 'MPS', 'ROCM'
backend = 'auto' # 'auto', 'openvino', 'onnxruntime', 'opencv'
Expand All @@ -93,7 +102,7 @@ output_format = 'openpose' # 'openpose', 'mmpose', 'deeplabcut', 'none' or a lis


[synchronization]
display_sync_plots = true # true or false (lowercase)
display_sync_plots = false # true or false (lowercase)
keypoints_to_consider = 'all' # 'all' if all points should be considered, for example if the participant did not perform any particicular sharp movement. In this case, the capture needs to be 5-10 seconds long at least
# ['RWrist', 'RElbow'] list of keypoint names if you want to specify keypoints with a sharp vertical motion.
approx_time_maxspeed = 'auto' # 'auto' if you want to consider the whole capture (default, slower if long sequences)
Expand Down
Loading
Loading