README.md 2.17 KB
Newer Older
yinchimaoliang's avatar
yinchimaoliang committed
1
### Prepare SUN RGB-D Data
2
We follow the procedure in [votenet](https://github.com/facebookresearch/votenet/).
yinchimaoliang's avatar
yinchimaoliang committed
3

liyinhao's avatar
liyinhao committed
4
1. Download SUNRGBD v2 data [HERE](http://rgbd.cs.princeton.edu/data/). Then, move SUNRGBD.zip, SUNRGBDMeta2DBB_v2.mat, SUNRGBDMeta3DBB_v2.mat and SUNRGBDtoolbox.zip to the OFFICIAL_SUNRGBD folder, unzip the zip files.
yinchimaoliang's avatar
yinchimaoliang committed
5

liyinhao's avatar
liyinhao committed
6
2. Enter the `matlab` folder, Extract point clouds and annotations by running `extract_split.m`, `extract_rgbd_data_v2.m` and `extract_rgbd_data_v1.m`.
yinchimaoliang's avatar
yinchimaoliang committed
7

8
3. Back to this directory, prepare data by running `python sunrgbd_data.py --gen_v1_data`.
liyinhao's avatar
liyinhao committed
9

10
11
12
13
4. Enter the project root directory, Generate training data by running
```bash
python tools/create_data.py sunrgbd --root-path ./data/sunrgbd --out-dir ./data/sunrgbd --extra-tag sunrgbd
```
yinchimaoliang's avatar
yinchimaoliang committed
14

liyinhao's avatar
liyinhao committed
15
16
17
18
19
20
21
22
23
24
25
The overall process could be achieved through the following script
```bash
cd matlab
matlab -nosplash -nodesktop -r 'extract_split;quit;'
matlab -nosplash -nodesktop -r 'extract_rgbd_data_v2;quit;'
matlab -nosplash -nodesktop -r 'extract_rgbd_data_v1;quit;'
cd ..
python sunrgbd_data.py --gen_v1_data
cd ..
python tools/create_data.py sunrgbd --root-path ./data/sunrgbd --out-dir ./data/sunrgbd --extra-tag sunrgbd
```
yinchimaoliang's avatar
yinchimaoliang committed
26
27

NOTE: SUNRGBDtoolbox.zip should have MD5 hash `18d22e1761d36352f37232cba102f91f` (you can check the hash with `md5 SUNRGBDtoolbox.zip` on Mac OS or `md5sum SUNRGBDtoolbox.zip` on Linux)
liyinhao's avatar
liyinhao committed
28

29
The directory structure after pre-processing should be as below
liyinhao's avatar
liyinhao committed
30
31
32
33
34
35
36
37
38
39
40
41
42
```
sunrgbd
├── sunrgbd_utils.py
├── sunrgbd_data.py
├── README.md
├── matlab
│   ├── extract_rgbd_data_v1.m
│   ├── extract_rgbd_data_v2.m
│   ├── extract_split.m
├── OFFICIAL_SUNRGBD
│   ├── SUNRGBD
│   ├── SUNRGBDMeta2DBB_v2.mat
│   ├── SUNRGBDMeta3DBB_v2.mat
liyinhao's avatar
liyinhao committed
43
│   ├── SUNRGBDtoolbox
44
45
46
47
48
49
50
51
52
53
54
55
56
57
├── sunrgbd_trainval
│   ├── calib
│   ├── image
│   ├── label_v1
│   ├── train_data_idx.txt
│   ├── depth
│   ├── label
│   ├── seg_label
│   ├── val_data_idx.txt
├── sunrgbd_pc_bbox_votes_50k_v1_train
├── sunrgbd_pc_bbox_votes_50k_v1_val
├── points
├── sunrgbd_infos_train.pkl
├── sunrgbd_infos_val.pkl
liyinhao's avatar
liyinhao committed
58
59

```