Mercurial Hosting > traffic-intelligence

comparison scripts/learn-motion-patterns.py @ 1012:01db14e947e4

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
resolved
author Wendlasida
date Fri, 01 Jun 2018 10:47:49 -0400
parents 933670761a57
children cc5cb04b04b0
comparison
equal deleted inserted replaced
1011:4f0312bee393 1012:01db14e947e4
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