Update extract_smplx_pose.py

extract_smplx_pose.py

@@ -33,7 +33,7 @@ import logging
 logging.getLogger("torch.utils._sympy.interp").setLevel(logging.ERROR)
 logging.getLogger("torch._inductor.utils").setLevel(logging.ERROR)
 
-PROJECT_ROOT = os.path.abspath(r"/mnt/shared-storage-user/mllm/zangyuhang/pmx/SMPL-X_pose_extraction")
+PROJECT_ROOT = os.path.abspath(r"/mnt/shared-storage-user/mllm/pmx/SMPL-X_pose_extraction")
 SMPLEST_X_PATH = os.path.join(PROJECT_ROOT, 'SMPLest-X')
 WILOR_PATH = os.path.join(PROJECT_ROOT, 'WiLoR')
 DEPTH_ANYTHING_PATH = os.path.join(PROJECT_ROOT, 'Depth-Anything-V2')
@@ -649,9 +649,9 @@ if __name__ == "__main__":
 
     parser.add_argument('--apply_sr', action='store_true', default=False)
     parser.add_argument('--opt_depth', action='store_true', default=False)
-    parser.add_argument('--yolo_ckpt', type=str, default=r'/mnt/shared-storage-user/mllm/zangyuhang/pmx/SMPL-X_pose_extraction/pretrained_weight/yolo/yolo26l.pt')
-    parser.add_argument('--hand_detector_ckpt', type=str, default=r'/mnt/shared-storage-user/mllm/zangyuhang/pmx/SMPL-X_pose_extraction/pretrained_weight/wilor/detector.pt')
-    parser.add_argument('--wilor_ckpt', type=str, default=r'/mnt/shared-storage-user/mllm/zangyuhang/pmx/SMPL-X_pose_extraction/pretrained_weight/wilor/wilor_final.ckpt')
-    parser.add_argument('--real_esrgan_ckpt', type=str, default=r'/mnt/shared-storage-user/mllm/zangyuhang/pmx/SMPL-X_pose_extraction/pretrained_weight/realesrgan/RealESRGAN_x4plus.pth')
-    parser.add_argument('--depth_anything_v2_ckpt', type=str, default=r'/mnt/shared-storage-user/mllm/zangyuhang/pmx/SMPL-X_pose_extraction/pretrained_weight/depth_anything-v2/depth_anything_v2_vitl.pth')
+    parser.add_argument('--yolo_ckpt', type=str, default=r'/mnt/shared-storage-user/mllm/pmx/SMPL-X_pose_extraction/pretrained_weight/yolo/yolo26l.pt')
+    parser.add_argument('--hand_detector_ckpt', type=str, default=r'/mnt/shared-storage-user/mllm/pmx/SMPL-X_pose_extraction/pretrained_weight/wilor/detector.pt')
+    parser.add_argument('--wilor_ckpt', type=str, default=r'/mnt/shared-storage-user/mllm/pmx/SMPL-X_pose_extraction/pretrained_weight/wilor/wilor_final.ckpt')
+    parser.add_argument('--real_esrgan_ckpt', type=str, default=r'/mnt/shared-storage-user/mllm/pmx/SMPL-X_pose_extraction/pretrained_weight/realesrgan/RealESRGAN_x4plus.pth')
+    parser.add_argument('--depth_anything_v2_ckpt', type=str, default=r'/mnt/shared-storage-user/mllm/pmx/SMPL-X_pose_extraction/pretrained_weight/depth_anything-v2/depth_anything_v2_vitl.pth')
     main(parser.parse_args())