Mercurial Hosting > traffic-intelligence
diff scripts/learn-motion-patterns.py @ 998:933670761a57
updated code to python 3 (tests pass and scripts run, but non-executed parts of code are probably still not correct)
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Sun, 27 May 2018 23:22:48 -0400 |
parents | 23f98ebb113f |
children | cc5cb04b04b0 |
line wrap: on
line diff
--- a/scripts/learn-motion-patterns.py Fri May 25 18:15:18 2018 -0400 +++ b/scripts/learn-motion-patterns.py Sun May 27 23:22:48 2018 -0400 @@ -1,4 +1,4 @@ -#! /usr/bin/env python +#! /usr/bin/env python3 import sys, argparse @@ -61,7 +61,7 @@ initialPrototypes = storage.loadPrototypesFromSqlite(args.inputPrototypeDatabaseFilename, True) trajectories = [p.getMovingObject().getPositions().asArray().T for p in initialPrototypes]+trajectories if len(initialPrototypes) > 0: - initialPrototypeIndices = range(len(initialPrototypes)) + initialPrototypeIndices = list(range(len(initialPrototypes))) else: initialPrototypeIndices = None else: