Skip to content

Commit

Permalink
Workaround to fix egopose visualization bug
Browse files Browse the repository at this point in the history
  • Loading branch information
holger-motional committed Oct 25, 2018
1 parent 85db2a1 commit 2ab5aa0
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion python-sdk/nuscenes_utils/nuscenes.py
Original file line number Diff line number Diff line change
Expand Up @@ -838,9 +838,14 @@ def render_egoposes_on_map(self, log_location: str, scene_tokens: List=None, dem
# Settings
close_dist = 100
pixel_to_meter = 0.1
ignore_logfiles = ['n008-2018-05-21-11-06-59-0400'] # Exclude older logs with incompatible maps

log_tokens = [l['token'] for l in self.nusc.log if l['location'] == log_location]
# Get logs by location
log_tokens = [l['token'] for l in self.nusc.log if l['location'] == log_location
and l['logfile'] not in ignore_logfiles]
assert len(log_tokens) > 0

# Filter scenes
scene_tokens_location = [e['token'] for e in self.nusc.scene if e['log_token'] in log_tokens]
if scene_tokens is not None:
scene_tokens_location = [t for t in scene_tokens_location if t in scene_tokens]
Expand Down

0 comments on commit 2ab5aa0

Please sign in to comment.