wkentaro / morefusion

MoreFusion: Multi-object Reasoning for 6D Pose Estimation from Volumetric Fusion, CVPR 2020
https://morefusion.wkentaro.com
Other
230 stars 47 forks source link

lack of dataset #69

Closed IzumiKUN0905 closed 3 years ago

IzumiKUN0905 commented 3 years ago

Hello, I would like to train the 6d pose prediction model and I can't find the dataset "20190916_124002.877532.v2.reindexed".

(base) robotdl@robotdl:~/morefusion-master/examples/ycb_video/singleview_pcd$ ./download_dataset.py Computing MD5: /home/robotdl/data/ycb_video/YCB_Video_Models.zip MD5 matches: /home/robotdl/data/ycb_video/YCB_Video_Models.zip Traceback (most recent call last): File "./download_dataset.py", line 11, in split="train" File "/home/robotdl/morefusion-master/morefusion/datasets/rgbd_pose_estimation/my_synthetic_ycb20190916/reindexed.py", line 27, in init split=split, class_ids=class_ids, augmentation=augmentation, File "/home/robotdl/morefusion-master/morefusion/datasets/rgbd_pose_estimation/reindexed.py", line 16, in init raise IOError(f"{self.root_dir} does not exist. ") OSError: /home/robotdl/data/wkentaro/morefusion/ycb_video/synthetic_data/20190916_124002.877532.v2.reindexed does not exist.

I just found a similar one named "20190916_124002.877532.reindexed" but it doesn't have a meta.joson file inside. So, where can I download the dataset? Thank you.

wkentaro commented 3 years ago

Ah, I'll fix this soon. You can download the v2.reindexed from here: https://drive.google.com/u/1/uc?id=1StTkkkKtgVffo8vCr-gW7tFvfi_FzlHX

IzumiKUN0905 commented 3 years ago

I'm training on your v2 dataset and meeting a new problem. I find that when the split is "syn", the parameter sampling will not be set(see line 30 to 48 in the file ".../morefusion-master/morefusion/datasets/rgbd_pose_estimation/ycb_video/reindexed.py"). It will make the program crash in the random_sampling.py. So I just assign a value of 8 to the sampling when the split is "syn". Is that right?