search

dingdingcai / GSPose

MIT License
70 stars 6 forks source link

can’t find the file #7

Open 2021270902001sc opened 4 months ago

2021270902001sc commented 4 months ago

I have downloaded part of the megapose-gso-1M dataset. And the python dataset/extract_megapose_to_BOP.py was executed, but I couldn't find valid_meshes.json in MegaPose. Code: training/training.py:50-57 DATA_DIR = os.path.join(PROJ_ROOT, 'dataspace', 'MegaPose')

dataset = Dataset(data_dir=DATA_DIR, query_view_num=que_view_num, refer_view_num=refer_view_num, rand_view_num=random_view_num, nnb_Rmat_threshold=nnb_Rmat_threshold, ) dataset/megapose_dataset.py:81-85

    cad_model_dir = os.path.join(self.data_dir, 'models')
    valid_mesh_names = mmcv.load(os.path.join(self.data_dir, 'valid_meshes.json'))
    valid_mesh_info_path = os.path.join(self.data_dir, 'valid_meshes_diameters.json')
    valid_mesh_bbox3D_path = os.path.join(self.data_dir, 'valid_meshes_3D_bboxes.json')

valid_mesh_names Is dataspace/MegaPose/valid_meshes.json，But I can't find this file.

binbin0801 commented 4 months ago

Have you solved this problem?

2021270902001sc commented 4 months ago

No,I find dataspace/MegaPose/google_scanned_objects/valid_meshes.json, but google_scanned_objects doesn't have cad_model_dir(models) The train_pbr which I have doesn't have any .json,I guess the .json is created by other files.

7-Ryan commented 3 months ago

Have you solved the problem? Can you complete the code train python training/training.py?