j96w / DexCap

[RSS 2024] "DexCap: Scalable and Portable Mocap Data Collection System for Dexterous Manipulation" code repository
MIT License
216 stars 24 forks source link

How to deploy #1

Open kassajing opened 6 months ago

kassajing commented 6 months ago

I trained the model using the 1-20 data. How should I deploy it to a real robotic arm?

j96w commented 6 months ago

Hi @kassajing, glad that you have successfully gone through the data processing and training parts!! The learned policy outputs 46-dim action space consisting of (in order):

You will need a real robot to verify the learned policy. The LEAP hand is open-sourced and can be built by following the LEAP manual. You can use the robot arm you have, and use end-effector position control with the 7-dim translation and orientation outputs.

kassajing commented 6 months ago

Thank you very much for your reply!I have a new question need your help. Regarding the """....layers.0.weight" and """...layers.0.bias" in the model, the dimension of bias is (32,) (64,) (128,) (256, ), does this part correspond to the joints of the 16-dim LEAP hand? Does (32,6,1) (64,,32,1) (128,64,1) (256,128,1) in weights correspond to the arm? Which dimension of data corresponds to 3-dim or 4-dim?

j96w commented 5 months ago

I think you are referring to the weights of the MLP layers, which are not the final action outputs. Checkout this function to see how the action is generated. The action format (46-dim action output) aligns with the action output in this function.