search

JiehongLin / SAM-6D

[CVPR2024] Code for "SAM-6D: Segment Anything Model Meets Zero-Shot 6D Object Pose Estimation".
373 stars 32 forks source link

Training on MegaPose Training Set #79

Open tyleeeeee opened 3 weeks ago

tyleeeeee commented 3 weeks ago

Hi,

I'm training the Pose Estimation Model of SAM-6D. I've prepared the MegaPose training data and rendered the template in Render folder. Then I ran this command in Pose Estimation Model folder

python train.py --gpus 0,1,2,3 --model pose_estimation_model --config config/base.yaml

However, an error occurs

FileNotFoundError: [Errno 2] No such file or directory: '../Data/MegaPose-Training-Data/MegaPose-GSO/train_pbr_web/key_to_shard.json'

I'm not sure where to find this JSON file.