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.

Sign up for GitHub

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

Aaron/openpose tracking #47

Merged
merged 30 commits into from
Aug 28, 2024
Merged

Aaron/openpose tracking #47

Changes from 1 commit
Commits
Show all changes
30 commits
Select commit Hold shift + click to select a range
85477ea
Philip/flesh out mediapipe model info (#46)
aaroncherian Jul 19, 2024
4b02577
fixing openpose imports into process_folder
aaroncherian Jul 19, 2024
67f6ce8
Update openpose_model_info.py
aaroncherian Jul 19, 2024
e190b01
Update process_folder_of_videos.py
aaroncherian Jul 19, 2024
91b9f2b
fix tracker name/model info argument
aaroncherian Jul 19, 2024
7c88cc8
Merge branch 'philip/flesh_out_mediapipe_model_info' into aaron/openp…
aaroncherian Jul 22, 2024
b3da0ab
Revert "Merge branch 'philip/flesh_out_mediapipe_model_info' into aar…
aaroncherian Jul 22, 2024
c96b3cd
fixing openpose exe path issues
aaroncherian Jul 22, 2024
47768aa
remove unused parser file
philipqueen Jul 22, 2024
4f41c7c
Merge branch 'aaron/openpose_tracking' of https://github.com/freemoca…
philipqueen Jul 22, 2024
df83b8c
change parameter locations and add hands and face options
philipqueen Jul 22, 2024
f36445d
add output_resolution as parameter and fill out tracking params
philipqueen Jul 22, 2024
e01cc3d
move tracking parameters back to initialization from process_video
philipqueen Jul 22, 2024
53496cc
fix some imports and type errors
philipqueen Jul 23, 2024
31798d2
fixing hands/face parameter bug
aaroncherian Jul 23, 2024
f33719f
handle optional hand and face data in recorder
philipqueen Jul 24, 2024
b19e997
cleanup example usage in tracker
philipqueen Jul 24, 2024
2e1529d
cleanup example usage in process_folder_of_videos
philipqueen Jul 24, 2024
89a85e1
remove unused imports in process_folder_of_videos
philipqueen Jul 24, 2024
e4943e2
fix type hinting in BaseTrackers
philipqueen Jul 24, 2024
188594e
fix problem in BaseRecorder
philipqueen Jul 24, 2024
ce400a2
update openpose model info with num tracked points of hands/face
aaroncherian Jul 24, 2024
38abdc8
fix hand and face landmark number references and remove json path fro…
philipqueen Jul 26, 2024
0065987
autoselect json output path if none is given
philipqueen Jul 26, 2024
c5823f9
fix output path example to avi instead of mp4
philipqueen Jul 26, 2024
1a503ea
check for missing frame error
philipqueen Jul 29, 2024
39072b6
pass face/hands bool to recorder/set default net res correctly
aaroncherian Jul 29, 2024
6288a4d
fix recorded tracked points bugs
aaroncherian Jul 29, 2024
0209c32
make output json path optional
aaroncherian Jul 29, 2024
c6dfbb1
fix for the example path change I accidentally pushed
aaroncherian Jul 29, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 5 additions & 1 deletion skellytracker/process_folder_of_videos.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -193,9 +193,11 @@ def get_tracker(tracker_name: str, tracking_params: BaseModel) -> BaseTracker:
elif tracker_name == 'OpenPoseTracker':
tracker = OpenPoseTracker(
openpose_root_folder_path=tracking_params.openpose_root_folder_path,
output_json_path=tracking_params.output_json_path,
output_json_folder_path=tracking_params.output_json_path,
net_resolution=tracking_params.net_resolution,
number_people_max=tracking_params.number_people_max,
track_faces=tracking_params.track_face,
track_hands=tracking_params.track_hands,
)

else:
Expand Down Expand Up @@ -259,6 +261,8 @@ def get_tracker_params(tracker_name: str) -> BaseModel:
tracking_params=OpenPoseTrackingParams(
openpose_root_folder_path=str(openpose_root_folder_path),
output_json_path=str(output_json_path),
track_hands=False,
philipqueen marked this conversation as resolved.
Show resolved Hide resolved
track_face=False
),
synchronized_video_path=input_video_filepath,
num_processes=num_processes,
Expand Down