3d visualization
Browse files
main.py
CHANGED
@@ -11,8 +11,7 @@ print("[INFO]: Imported modules!")
|
|
11 |
# inferencer = MMPoseInferencer('hand') # 'hand', 'human , device='cuda'
|
12 |
# inferencer = MMPoseInferencer('human')
|
13 |
|
14 |
-
inferencer = MMPoseInferencer(pose3d="
|
15 |
-
pose3d_weights="https://download.openmmlab.com/mmpose/body3d/simple_baseline/simple3Dbaseline_h36m-f0ad73a4_20210419.pth")
|
16 |
# https://github.com/open-mmlab/mmpose/tree/dev-1.x/configs/body_3d_keypoint/pose_lift
|
17 |
# motionbert_ft_h36m-d80af323_20230531.pth
|
18 |
# simple3Dbaseline_h36m-f0ad73a4_20210419.pth
|
@@ -62,7 +61,7 @@ def poses(photo):
|
|
62 |
# cv2.destroyAllWindows() # Closing window
|
63 |
output_file = glob.glob("*.mp4")
|
64 |
print(output_file)
|
65 |
-
|
66 |
return output_file[0]
|
67 |
|
68 |
|
|
|
11 |
# inferencer = MMPoseInferencer('hand') # 'hand', 'human , device='cuda'
|
12 |
# inferencer = MMPoseInferencer('human')
|
13 |
|
14 |
+
inferencer = MMPoseInferencer(pose3d="human3d")
|
|
|
15 |
# https://github.com/open-mmlab/mmpose/tree/dev-1.x/configs/body_3d_keypoint/pose_lift
|
16 |
# motionbert_ft_h36m-d80af323_20230531.pth
|
17 |
# simple3Dbaseline_h36m-f0ad73a4_20210419.pth
|
|
|
61 |
# cv2.destroyAllWindows() # Closing window
|
62 |
output_file = glob.glob("*.mp4")
|
63 |
print(output_file)
|
64 |
+
|
65 |
return output_file[0]
|
66 |
|
67 |
|