Skip to content

GitLab

Commit 4a1f5116 authored by liyinhao's avatar liyinhao Committed by zhangwenwei
Browse files

add prepare datasets in GETTING_STARTED.md

parent 3a9012e4
Show whitespace changes
Inline Side-by-side
Showing with 48 additions and 30 deletions
docs/GETTING_STARTED.md
View file @ 4a1f5116
...@@ -9,39 +9,57 @@ It is recommended to symlink the dataset root to `$MMDETECTION/data`. ...@@ -9,39 +9,57 @@ It is recommended to symlink the dataset root to `$MMDETECTION/data`.
If your folder structure is different, you may need to change the corresponding paths in config files. If your folder structure is different, you may need to change the corresponding paths in config files.
``` ```
mmdetection mmdetection3d
├── mmdet ├── mmdet3d
├── tools ├── tools
├── configs ├── configs
├── data ├── data
│ ├── coco │ ├── nuscenes
│ │ ├── annotations │ │ ├── maps
│ │ ├── train2017 │ │ ├── samples
│ │ ├── val2017 │ │ ├── sweeps
│ │ ├── test2017 │ │ ├── v1.0-test
│ ├── cityscapes | | ├── v1.0-trainval
│ │ ├── annotations │ ├── kitti
│ │ ├── leftImg8bit │ │ ├── ImageSets
│ │ │ ├── train │ │ ├── testing
│ │ │ ├── val │ │ │ ├── calib
│ │ ├── gtFine │ │ │ ├── image_2
│ │ │ ├── train │ │ │ ├── velodyne
│ │ │ ├── val │ │ ├── training
│ ├── VOCdevkit │ │ │ ├── calib
│ │ ├── VOC2007 │ │ │ ├── image_2
│ │ ├── VOC2012 │ │ │ ├── label_2
│ │ │ ├── velodyne
``` │ ├── scannet
│ │ ├── meta_data
The cityscapes annotations have to be converted into the coco format using `tools/convert_datasets/cityscapes.py`: │ │ ├── scans
│ │ ├── batch_load_scannet_data.py
```shell │ │ ├── load_scannet_data.py
pip install cityscapesscripts │ │ ├── scannet_utils.py
python tools/convert_datasets/cityscapes.py ./data/cityscapes --nproc 8 --out_dir ./data/cityscapes/annotations │ │ ├── README.md
``` │ ├── sunrgbd
│ │ ├── OFFICIAL_SUNRGBD
Currently the config files in `cityscapes` use COCO pre-trained weights to initialize. │ │ ├── matlab
You could download the pre-trained models in advance if network is unavailable or slow, otherwise it would cause errors at the beginning of training. │ │ ├── sunrgbd_data.py
│ │ ├── sunrgbd_utils.py
│ │ ├── README.md
```
Download nuScenes V1.0 full dataset data [HERE]( https://www.nuscenes.org/download). Prepare nuscenes data by running
```bash
python tools/create_data.py nuscenes --root-path ./data/nuscenes --out-dir ./data/nuscenes --extra-tag nuscenes
```
Download KITTI 3D detection data [HERE](http://www.cvlibs.net/datasets/kitti/eval_object.php?obj_benchmark=3d). Prepare kitti data by running
```bash
python tools/create_data.py kitti --root-path ./data/kitti --out-dir ./data/kitti --extra-tag kitti
```
To prepare scannet data, please see [scannet](../data/scannet/README.md).
To prepare sunrgbd data, please see [sunrgbd](../data/sunrgbd/README.md).
For using custom datasets, please refer to [Tutorials 2: Adding New Dataset](tutorials/new_dataset.md). For using custom datasets, please refer to [Tutorials 2: Adding New Dataset](tutorials/new_dataset.md).
......
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