Unverified Commit 043eb997 authored by Yezhen Cong's avatar Yezhen Cong Committed by GitHub
Browse files

[Doc] SUNRGBD dataset doc update (#391)

* sunrgbd doc update and change in image path

* rephrase
parent 6582c5e0
...@@ -22,6 +22,11 @@ python tools/create_data.py sunrgbd --root-path ./data/sunrgbd --out-dir ./data ...@@ -22,6 +22,11 @@ python tools/create_data.py sunrgbd --root-path ./data/sunrgbd --out-dir ./data
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) 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)
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
```bash
python tools/create_data.py sunrgbd --root-path ./data/sunrgbd --out-dir ./data/sunrgbd --extra-tag sunrgbd
```
The directory structure after pre-processing should be as below The directory structure after pre-processing should be as below
``` ```
sunrgbd sunrgbd
......
...@@ -48,7 +48,7 @@ class SUNRGBDDataset(Custom3DDataset): ...@@ -48,7 +48,7 @@ class SUNRGBDDataset(Custom3DDataset):
ann_file, ann_file,
pipeline=None, pipeline=None,
classes=None, classes=None,
modality=None, modality=dict(use_camera=True, use_lidar=True),
box_type_3d='Depth', box_type_3d='Depth',
filter_empty_gt=True, filter_empty_gt=True,
test_mode=False): test_mode=False):
...@@ -61,8 +61,8 @@ class SUNRGBDDataset(Custom3DDataset): ...@@ -61,8 +61,8 @@ class SUNRGBDDataset(Custom3DDataset):
box_type_3d=box_type_3d, box_type_3d=box_type_3d,
filter_empty_gt=filter_empty_gt, filter_empty_gt=filter_empty_gt,
test_mode=test_mode) test_mode=test_mode)
if self.modality is None: assert 'use_camera' in self.modality and \
self.modality = dict(use_camera=True, use_lidar=True) 'use_lidar' in self.modality
assert self.modality['use_camera'] or self.modality['use_lidar'] assert self.modality['use_camera'] or self.modality['use_lidar']
def get_data_info(self, index): def get_data_info(self, index):
...@@ -94,8 +94,9 @@ class SUNRGBDDataset(Custom3DDataset): ...@@ -94,8 +94,9 @@ class SUNRGBDDataset(Custom3DDataset):
input_dict['file_name'] = pts_filename input_dict['file_name'] = pts_filename
if self.modality['use_camera']: if self.modality['use_camera']:
img_filename = osp.join(self.data_root, img_filename = osp.join(
info['image']['image_path']) osp.join(self.data_root, 'sunrgbd_trainval'),
info['image']['image_path'])
input_dict['img_prefix'] = None input_dict['img_prefix'] = None
input_dict['img_info'] = dict(filename=img_filename) input_dict['img_info'] = dict(filename=img_filename)
calib = info['calib'] calib = info['calib']
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment