23 lines
483 B
Python
23 lines
483 B
Python
import pickle
|
|
|
|
# data is
|
|
# recording = {}
|
|
# recording["mujoco_vel"] = []
|
|
# recording["robot_vel"] = []
|
|
data = pickle.load(open("speeds.pkl", "rb"))
|
|
|
|
|
|
import matplotlib.pyplot as plt
|
|
|
|
mujoco_vel = data.get("mujoco_vel", [])
|
|
robot_vel = data.get("robot_vel", [])
|
|
|
|
plt.figure()
|
|
plt.plot(mujoco_vel, label="Mujoco Velocity")
|
|
plt.plot(robot_vel, label="Robot Velocity")
|
|
plt.xlabel("Time Step")
|
|
plt.ylabel("Velocity")
|
|
plt.title("Mujoco Velocity vs Robot Velocity")
|
|
plt.legend()
|
|
plt.show()
|