comparison python/storage.py @ 203:e2f31813ade6

added code to display trajectories on videa
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Tue, 06 Mar 2012 18:10:19 -0500
parents 319a04ba9033
children 966c2cd2bd9f
comparison
equal deleted inserted replaced
202:b0b964ba9489 203:e2f31813ade6
45 if obj: 45 if obj:
46 objects.append(obj) 46 objects.append(obj)
47 47
48 connection.close() 48 connection.close()
49 return objects 49 return objects
50
51 def loadObjectsFromSqlite(filename, objectNumbers = -1):
52 '''Loads objects as averages of feature trajectories
53 TODO: load features as well, other ways of averaging trajectories
54 need to provide table name(s) ?'''
55 # elect frame_number, avg(x_coordinate), avg(y_coordinate) from positions where trajectory_id in (select trajectory_id from objects_features where object_id=12) group by frame_number;
56
57
58
50 59
51 def loadTrajectoriesFromNgsimFile(filename, nObjects = -1, sequenceNum = -1): 60 def loadTrajectoriesFromNgsimFile(filename, nObjects = -1, sequenceNum = -1):
52 '''Reads data from the trajectory data provided by NGSIM project 61 '''Reads data from the trajectory data provided by NGSIM project
53 and returns the list of Feature objects''' 62 and returns the list of Feature objects'''
54 objects = [] 63 objects = []