Mercurial > hg > nsaunier > traffic-intelligence
diff trafficintelligence/processing.py @ 1074:b123fa0e5440
merged
| author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
|---|---|
| date | Fri, 20 Jul 2018 16:24:30 -0400 |
| parents | c67f8c36ebc7 |
| children |
line wrap: on
line diff
--- a/trafficintelligence/processing.py Fri Jul 20 16:23:56 2018 -0400 +++ b/trafficintelligence/processing.py Fri Jul 20 16:24:30 2018 -0400 @@ -18,16 +18,15 @@ objectsNotInZone.append(o) return speeds, objectsNotInZone -def extractVideoSequenceSpeeds(dbFilename, siteName, nObjects, startTime, frameRate, minUserDurationSeconds, aggMethods, aggCentiles): +def extractVideoSequenceSpeeds(dbFilename, siteName, nObjects, startTime, frameRate, minDuration, aggMethods, aggCentiles): data = [] d = startTime.date() t1 = startTime.time() - minUserDuration = minUserDurationSeconds*frameRate print('Extracting speed from '+dbFilename) aggFunctions, tmpheaders = utils.aggregationMethods(aggMethods, aggCentiles) objects = storage.loadTrajectoriesFromSqlite(dbFilename, 'object', nObjects) for o in objects: - if o.length() > minUserDuration: + if o.length() > minDuration: row = [siteName, d, utils.framesToTime(o.getFirstInstant(), frameRate, t1), o.getUserType()] tmp = o.getSpeeds() for method,func in aggFunctions.items():