Skip to content

Commit 0b3798f

Browse files
committed
max sweeps should be 9 to match nuscenes limit
1 parent 4af9541 commit 0b3798f

File tree

3 files changed

+5
-5
lines changed

3 files changed

+5
-5
lines changed

README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -132,8 +132,8 @@ Since the dataset is really large, you can download parts of the dataset.
132132

133133
Then run
134134
```bash
135-
python create_data.py nuscenes_data_prep --data_path=NUSCENES_TRAINVAL_DATASET_ROOT --version="v1.0-trainval" --max_sweeps=10
136-
python create_data.py nuscenes_data_prep --data_path=NUSCENES_TEST_DATASET_ROOT --version="v1.0-test" --max_sweeps=10
135+
python create_data.py nuscenes_data_prep --data_path=NUSCENES_TRAINVAL_DATASET_ROOT --version="v1.0-trainval" --max_sweeps=9
136+
python create_data.py nuscenes_data_prep --data_path=NUSCENES_TEST_DATASET_ROOT --version="v1.0-test" --max_sweeps=9
137137
```
138138

139139
* Modify config file

second/create_data.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ def kitti_data_prep(root_path):
1313
kitti_ds.create_reduced_point_cloud(root_path)
1414
create_groundtruth_database("KittiDataset", root_path, Path(root_path) / "kitti_infos_train.pkl")
1515

16-
def nuscenes_data_prep(root_path, version, max_sweeps=10):
16+
def nuscenes_data_prep(root_path, version, max_sweeps=9):
1717
nu_ds.create_nuscenes_infos(root_path, version=version, max_sweeps=max_sweeps)
1818
create_groundtruth_database("NuScenesDataset", root_path, Path(root_path) / "infos_train.pkl")
1919

second/data/nuscenes_dataset.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -434,7 +434,7 @@ def _fill_trainval_infos(nusc,
434434
train_scenes,
435435
val_scenes,
436436
test=False,
437-
max_sweeps=10):
437+
max_sweeps=9):
438438
train_nusc_infos = []
439439
val_nusc_infos = []
440440
from pyquaternion import Quaternion
@@ -523,7 +523,7 @@ def _fill_trainval_infos(nusc,
523523
return train_nusc_infos, val_nusc_infos
524524

525525

526-
def create_nuscenes_infos(root_path, version="v1.0-trainval", max_sweeps=10):
526+
def create_nuscenes_infos(root_path, version="v1.0-trainval", max_sweeps=9):
527527
from nuscenes.nuscenes import NuScenes
528528
nusc = NuScenes(version=version, dataroot=root_path, verbose=True)
529529
from nuscenes.utils import splits

0 commit comments

Comments
 (0)