search

jeffsonyu / VTacO

Visual-Tactile Sensing for In-Hand Object Reconstruction
MIT License
28 stars 1 forks source link

网络的训练权重 #7

Closed meiguiz closed 4 weeks ago

meiguiz commented 2 months ago

您好！打扰了，我想要直接使用该项目，因为我并没有好的GPU进行训练。请问可以提供您已经训练完的权重文件嘛？万分感谢！包括python train_depth.py configs/tactile/tactile_test.yaml

jeffsonyu commented 2 months ago

Hi! The img-to-depth model checkpoint can be found in the following link. Thank you for your attention on our work! https://drive.google.com/file/d/1bWg-avRw2yLtPrSiCJ0e-ZtKrh21vAmE/view?usp=drive_link

meiguiz commented 2 months ago

在VTacO_YCB.yaml文件里，encoder_hand是不是应该设置为False，在encoder_t2d下面的encoder_hand是不是同样应该设置为False，现在的VTacO_YCB.yaml的这两个设置均为True

jeffsonyu commented 2 months ago

Actually, the param encoder_hand can be set to False since we don't predict the hand param in VTacO pipeline. However, the encoder_hand under the encoder_t2d_kwargs should set to True, since we train the model_depth.pth with a hand param predictor. You can try to load the checkpoint with the encoder_hand True or False

jeffsonyu commented 4 weeks ago

Close due to no update