Guest User

fk_calc.py

a guest
Dec 11th, 2024
22
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.05 KB | None | 0 0
  1. import adam
  2. from adam.pytorch import KinDynComputations
  3. import numpy as np
  4. import torch
  5.  
  6.  
  7. model_path = "/samsung4tb/franka_description/urdfs/fr3_franka_hand.urdf"
  8.  
  9. # joints_name_list = ["base_joint", "fr3_joint1", "fr3_joint2", "fr3_joint3", "fr3_joint4", "fr3_joint5", "fr3_joint6", "fr3_joint7", "fr3_joint8", "fr3_hand_joint", "fr3_hand_tcp_joint", "fr3_finger_joint1", "fr3_finger_joint2"]
  10.  
  11. joints_name_list = ["fr3_joint1", "fr3_joint2", "fr3_joint3", "fr3_joint4", "fr3_joint5", "fr3_joint6", "fr3_joint7"]
  12.  
  13. kinDyn = KinDynComputations(model_path, joints_name_list, root_link="base")
  14.  
  15. # you need to set it manually like this to work
  16. kinDyn.rbdalgos.root_link= "fr3_hand"
  17. # kinDyn = KinDynComputations(model_path, joints_name_list, root_link="fr3_hand")
  18.  
  19. camera_calib_home = torch.Tensor(
  20. [-0.19587807, 0.12579048, 0.72064054, -2.2718847 , 0.427838 ,
  21. 1.7621214 , -1.662231]
  22. )
  23.  
  24. out1 = kinDyn.forward_kinematics('fr3_hand', np.eye(4), camera_calib_home)
  25.  
  26. out2 = kinDyn.forward_kinematics('fr3_hand_tcp', np.eye(4), camera_calib_home)
Add Comment
Please, Sign In to add comment