Mercurial Hosting > traffic-intelligence

comparison scripts/learn-motion-patterns.py @ 998:933670761a57

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
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
comparison
equal deleted inserted replaced
997:4f3387a242a1 998:933670761a57
1 #! /usr/bin/env python 1 #! /usr/bin/env python3
2 2
3 import sys, argparse 3 import sys, argparse
4 4
5 #import matplotlib.pyplot as plt 5 #import matplotlib.pyplot as plt
6 import numpy as np 6 import numpy as np
59 # load initial prototypes, if any 59 # load initial prototypes, if any
60 if args.inputPrototypeDatabaseFilename is not None: 60 if args.inputPrototypeDatabaseFilename is not None:
61 initialPrototypes = storage.loadPrototypesFromSqlite(args.inputPrototypeDatabaseFilename, True) 61 initialPrototypes = storage.loadPrototypesFromSqlite(args.inputPrototypeDatabaseFilename, True)
62 trajectories = [p.getMovingObject().getPositions().asArray().T for p in initialPrototypes]+trajectories 62 trajectories = [p.getMovingObject().getPositions().asArray().T for p in initialPrototypes]+trajectories
63 if len(initialPrototypes) > 0: 63 if len(initialPrototypes) > 0:
64 initialPrototypeIndices = range(len(initialPrototypes)) 64 initialPrototypeIndices = list(range(len(initialPrototypes)))
65 else: 65 else:
66 initialPrototypeIndices = None 66 initialPrototypeIndices = None
67 else: 67 else:
68 initialPrototypes = [] 68 initialPrototypes = []
69 initialPrototypeIndices = None 69 initialPrototypeIndices = None