Skip to content

Commit

Permalink
Split train set into train_detect and train_track (nutonomy#212)
Browse files Browse the repository at this point in the history
* Split train set into train_track and train_detect, no modifications to train, cleanup

* Using stable version of miniconda Docker to avoid errors

* Clarification on different splits and added assertion
  • Loading branch information
holger-motional authored Aug 26, 2019
1 parent ac347b3 commit a695866
Show file tree
Hide file tree
Showing 4 changed files with 106 additions and 112 deletions.
6 changes: 3 additions & 3 deletions python-sdk/nuscenes/eval/detection/loaders.py
Original file line number Diff line number Diff line change
Expand Up @@ -56,9 +56,9 @@ def load_gt(nusc, eval_split: str, verbose: bool = False) -> EvalBoxes:
# Only keep samples from this split.
splits = create_splits_scenes()

# Check compatibility of split with nusc versions
# Check compatibility of split with nusc_version.
version = nusc.version
if eval_split in {'train', 'val'}:
if eval_split in {'train', 'val', 'train_detect', 'train_track'}:
assert version.endswith('trainval'), \
'Error: Requested split {} which is not compatible with NuScenes version {}'.format(eval_split, version)
elif eval_split in {'mini_train', 'mini_val'}:
Expand All @@ -72,7 +72,7 @@ def load_gt(nusc, eval_split: str, verbose: bool = False) -> EvalBoxes:
.format(eval_split))

if eval_split == 'test':
# Check that you aren't trying to cheat :)
# Check that you aren't trying to cheat :).
assert len(nusc.sample_annotation) > 0, \
'Error: You are trying to evaluate on the test set but you do not have the annotations!'

Expand Down
Loading

0 comments on commit a695866

Please sign in to comment.