add policy from local path

examples/unitree_g1/holosoma_locomotion.py

@@ -143,6 +143,7 @@ DEFAULT_HOLOSOMA_REPO_ID = "nepyope/holosoma_locomotion"
 def load_holosoma_policy(
     repo_id: str = DEFAULT_HOLOSOMA_REPO_ID,
     policy_name: str = "fastsac",
+    local_path: str | None = None
 ) -> ort.InferenceSession:
     """Load Holosoma 29-DOF locomotion policy from Hugging Face Hub.
 
@@ -150,6 +151,9 @@ def load_holosoma_policy(
         repo_id: Hugging Face Hub repository ID containing the ONNX policies.
         policy_name: Policy variant to load ("fastsac" or "ppo").
     """
+    if local_path is not None:
+        logger.info(f"Loading policy from local path: {local_path}")
+        policy_path = local_path
     filename_map = {
         "fastsac": "fastsac_g1_29dof.onnx",
         "ppo": "ppo_g1_29dof.onnx",
@@ -220,14 +224,6 @@ class HolosomaLocomotionController:
         logger.info(f"  Observation dim: 100, Action dim: 29")
         logger.info(f"  Missing joints (G1 23-DOF): {MISSING_JOINTS}")
 
-    def _transform_imu_data(self, waist_yaw, waist_yaw_omega, imu_quat, imu_omega):
-        """Transform IMU data from torso to pelvis frame."""
-        RzWaist = R.from_euler("z", waist_yaw).as_matrix()
-        R_torso = R.from_quat([imu_quat[1], imu_quat[2], imu_quat[3], imu_quat[0]]).as_matrix()
-        R_pelvis = np.dot(R_torso, RzWaist.T)
-        w = np.dot(RzWaist, imu_omega) - np.array([0, 0, waist_yaw_omega])
-        return R.from_matrix(R_pelvis).as_quat()[[3, 0, 1, 2]], w
-
     def holosoma_locomotion_run(self):
         """29-DOF whole-body locomotion policy loop - controls ALL 29 joints."""
         self.counter += 1
@@ -467,6 +463,12 @@ if __name__ == "__main__":
         choices=["fastsac", "ppo"],
         help="Policy variant to load (default: fastsac)",
     )
+    parser.add_argument(
+    "--local-path",
+    type=str,
+    default=None,
+    help="Path to local ONNX file (overrides --repo-id and --policy)",
+)
     args = parser.parse_args()
 
     # Load policy