yewsiang / Transferable3D

Transferable Semi-supervised 3D Object Detection from RGB-D Data, ICCV 2019
Apache License 2.0
22 stars 4 forks source link

Unable to find some .txt files for sunrgbd dataset #3

Open pawan2179 opened 3 years ago

pawan2179 commented 3 years ago

Hello Sir, I was trying to implement the model for predicting 3D bounding boxes. But when I tried to run sunrgbd_data.py I got the following error -

File "sunrgbd_data/sunrgbd_data.py", line 134, in extract_roi_seg data_idx_list = [int(line.rstrip()) for line in open(idx_filename)] FileNotFoundError: [Errno 2] No such file or directory: '/home/pawan/Downloads/Prism/Transferable3D-master/dataset/mysunrgbd/training/train_mini_data_idx.txt'

When I looked into the files these .txt files were not their

`

else:
    # Train on train, Test on val
    extract_roi_seg(pjoin(SUNRGBD_DATASET_DIR,'training','train_mini_data_idx.txt'), 'training',  
        output_filename=pjoin(BASE_DIR,'..','frustums','train_mini.zip.pickle'), viz=False, augmentX=1)
    extract_roi_seg(pjoin(SUNRGBD_DATASET_DIR,'training','train_data_idx.txt'), 'training',  
        output_filename=pjoin(BASE_DIR,'..','frustums','train_aug5x.zip.pickle'), viz=False, augmentX=5) 
    extract_roi_seg(pjoin(SUNRGBD_DATASET_DIR,'training','val_data_idx.txt'), 'training',
        output_filename=pjoin(BASE_DIR,'..','frustums','val.zip.pickle'), viz=False, augmentX=1)

    # Train on trainval, Test on test 
    extract_roi_seg(pjoin(SUNRGBD_DATASET_DIR,'training','trainval_data_idx.txt'), 'training',
        output_filename=pjoin(BASE_DIR,'..','frustums','trainval_aug5x.zip.pickle'), viz=False, augmentX=5)
    extract_roi_seg(pjoin(SUNRGBD_DATASET_DIR,'training','test_data_idx.txt'), 'training',
        output_filename=pjoin(BASE_DIR,'..','frustums','test.zip.pickle'), viz=False, augmentX=1)`

I'm a student and I'm not being able to figure out how to proceed. I would be really grateful if you could guide me a bit on how to proceed or tell me how can I get those files.

Thankyou

Mr-doraemon commented 2 years ago

Have you solved this problem? If you are, please tell me. Thank you very much.

JiachengDeng commented 2 years ago

You guys can solve this problem by trying the data prepare method of mmdetection3D like following https://mmdetection3d.readthedocs.io/zh_CN/latest/datasets/sunrgbd_det.html. Then you can find the train_data_idx.txt, val_data_idx.txt which are just two .txt file containing the idx of training data and val data respectively. For train_mini_data_idx.txt, you can create a .txt file with any samples you want.