rafaelcaricio/Open_Duck_Mini_Runtime
1
0
Fork 0
mirror of https://github.com/apirrone/Open_Duck_Mini_Runtime.git synced 2025-09-02 11:13:55 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

stuff

Browse source
This commit is contained in:
apirrone 2025-08-05 14:16:18 +02:00
parent de99f07477
commit 8433892a20
1 changed files with 8 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
scripts/v2_rl_walk_mujoco.py
View file

@ -129,6 +129,7 @@ class RLWalk:
print("IMU data is None, skipping observation retrieval")
return None
gravity = np.array(imu_data["orientation"]).reshape((3, 3)).T @ np.array([0, 0, -1])
# accelero = np.array(imu_data["accelero"])
gyro = imu_data["gyro"]
dof_pos = self.hwi.get_present_positions(
@ -162,6 +163,7 @@ class RLWalk:
obs = np.concatenate(
[
# accelero,
gyro,
gravity,
cmds,
@ -261,12 +263,12 @@ class RLWalk:
if obs is None:
continue
if np.linalg.norm(self.last_commands[:3]) > 0.1:
# if np.linalg.norm(self.last_commands[:3]) > 0.1:
self.imitation_i += 1 * (
self.phase_frequency_factor + self.phase_frequency_factor_offset
)
else:
self.imitation_i = 0
# else:
# self.imitation_i = 0
self.imitation_i = self.imitation_i % self.PRM.nb_steps_in_period
self.imitation_phase = np.array(
[