Skip to content

Commit 55e7185

Browse files
committed
fix wrong readme location
1 parent 7522f2f commit 55e7185

File tree

4 files changed

+17
-338
lines changed

4 files changed

+17
-338
lines changed
File renamed without changes.

README.md

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,8 @@ SECOND detector.
33

44
ONLY support python 3.6+, pytorch 1.0.0+. Tested in Ubuntu 16.04/18.04/Windows 10.
55

6+
If you want to train nuscenes dataset, see [this](NUSCENES-GUIDE.md).
7+
68
## News
79

810
2019-4-1: SECOND V1.6.0alpha released: New Data API, [NuScenes](https://www.nuscenes.org) support, [PointPillars](https://github.com/nutonomy/second.pytorch) support, fp16 and multi-gpu support.
@@ -44,6 +46,15 @@ bev AP:90.38, 88.20, 86.98
4446
3d AP:89.16, 78.78, 77.41
4547
```
4648

49+
### Performance in NuScenes validation set (NuScenes mini train set)
50+
51+
```
52+
car Nusc dist AP@0.5, 1.0, 2.0, 4.0
53+
62.80, 73.30, 76.85, 78.87
54+
pedestrian Nusc dist AP@0.5, 1.0, 2.0, 4.0
55+
61.09, 62.20, 63.66, 65.89
56+
```
57+
4758
## Install
4859

4960
### 1. Clone code
@@ -149,6 +160,7 @@ train_input_reader: {
149160
...
150161
}
151162
dataset: {
163+
dataset_class_name: "DATASET_NAME"
152164
kitti_info_path: "/path/to/dataset_infos_train.pkl"
153165
kitti_root_path: "DATASET_ROOT"
154166
}
@@ -157,6 +169,7 @@ train_input_reader: {
157169
eval_input_reader: {
158170
...
159171
dataset: {
172+
dataset_class_name: "DATASET_NAME"
160173
kitti_info_path: "/path/to/dataset_infos_val.pkl"
161174
kitti_root_path: "DATASET_ROOT"
162175
}
@@ -183,6 +196,10 @@ Assume you have 4 GPUs and want to train with 3 GPUs:
183196
CUDA_VISIBLE_DEVICES=0,1,3 python ./pytorch/train.py train --config_path=./configs/car.fhd.config --model_dir=/path/to/model_dir --multi_gpu=True
184197
```
185198

199+
Note: The batch_size and num_workers in config file is per-GPU, if you use multi-gpu, they will be multiplied by number of GPUs. Don't modify them manually.
200+
201+
You need to modify total step in config file. For example, 50 epochs = 15500 steps for car.lite.config and single GPU, if you use 4 GPUs, you need to divide ```steps``` and ```steps_per_eval``` by 4.
202+
186203
#### train with fp16 (mixed precision)
187204

188205
Modify config file, set enable_mixed_precision to true.

second/README.md

Lines changed: 0 additions & 289 deletions
This file was deleted.

second/RELEASE.md

Lines changed: 0 additions & 49 deletions
This file was deleted.

0 commit comments

Comments
 (0)