README.md 2.33 KB
Newer Older
yinchimaoliang's avatar
yinchimaoliang committed
1
### Prepare SUN RGB-D Data
Ziyi Wu's avatar
Ziyi Wu committed
2

3
We follow the procedure in [votenet](https://github.com/facebookresearch/votenet/).
yinchimaoliang's avatar
yinchimaoliang committed
4

liyinhao's avatar
liyinhao committed
5
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
6

liyinhao's avatar
liyinhao committed
7
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
8

9
3. Enter the project root directory, Generate training data by running
Ziyi Wu's avatar
Ziyi Wu committed
10

11
12
13
```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
The overall process could be achieved through the following script
Ziyi Wu's avatar
Ziyi Wu committed
16

liyinhao's avatar
liyinhao committed
17
18
19
20
21
```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;'
22
23
cd ../../..
python tools/create_data.py sunrgbd --root-path ./data/sunrgbd  --out-dir ./data/sunrgbd --extra-tag sunrgbd
liyinhao's avatar
liyinhao committed
24
```
yinchimaoliang's avatar
yinchimaoliang committed
25
26

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
27

28
NOTE: If you would like to play around with [ImVoteNet](../../configs/imvotenet/README.md), the image data (`./data/sunrgbd/sunrgbd_trainval/image`) are required. If you pre-processed the data before mmdet3d version 0.12.0, please pre-process the data again due to some updates in data pre-processing
Ziyi Wu's avatar
Ziyi Wu committed
29

30
31
32
33
```bash
python tools/create_data.py sunrgbd --root-path ./data/sunrgbd  --out-dir ./data/sunrgbd --extra-tag sunrgbd
```

34
The directory structure after pre-processing should be as below
Ziyi Wu's avatar
Ziyi Wu committed
35

liyinhao's avatar
liyinhao committed
36
37
38
39
40
41
42
43
44
45
46
```
sunrgbd
├── 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
47
│   ├── SUNRGBDtoolbox
48
49
50
51
52
53
54
55
56
57
58
59
├── sunrgbd_trainval
│   ├── calib
│   ├── image
│   ├── label_v1
│   ├── train_data_idx.txt
│   ├── depth
│   ├── label
│   ├── seg_label
│   ├── val_data_idx.txt
├── points
├── sunrgbd_infos_train.pkl
├── sunrgbd_infos_val.pkl
liyinhao's avatar
liyinhao committed
60
61

```