Mercurial Hosting > traffic-intelligence
changeset 1216:8356e15fd691
fix for result files
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Sun, 07 May 2023 16:24:27 -0400 |
parents | 1b472cddf9b1 |
children | 5038c357b57f |
files | trafficintelligence/storage.py |
diffstat | 1 files changed, 3 insertions(+), 1 deletions(-) [+] |
line wrap: on
line diff
--- a/trafficintelligence/storage.py Fri May 05 22:52:25 2023 -0400 +++ b/trafficintelligence/storage.py Sun May 07 16:24:27 2023 -0400 @@ -1299,7 +1299,7 @@ # from https://github.com/utiasSTARS/pykitti/blob/master/pykitti/utils.py -def loadTrajectoriesFromKITTI(filename, kittiCalibration, oxts): +def loadTrajectoriesFromKITTI(filename, kittiCalibration, oxts, resultFile = False): '''Reads data from KITTI ground truth or output from an object detection and tracking method kittiCalibration is obtained from loading training/testing calibration file for each sequence 00XX.txt @@ -1328,6 +1328,8 @@ 'l', # box length (in meters) 'x', 'y', 'z', # location (x,y,z) in camera coord 'ry'] # yaw angle (around Y-axis in camera coordinates) [-pi..pi] + if resultFile: + header.append('score') data = read_csv(filename, delimiter = ' ', names = header) data = data[data.trackingid > -1]