Mercurial Hosting > traffic-intelligence
diff scripts/learn-motion-patterns.py @ 919:7b3f2e0a2652
saving and loading prototype trajectories
| author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
|---|---|
| date | Wed, 05 Jul 2017 13:16:47 -0400 |
| parents | 89cc05867c4c |
| children | 499154254f37 |
line wrap: on
line diff
--- a/scripts/learn-motion-patterns.py Wed Jul 05 12:19:59 2017 -0400 +++ b/scripts/learn-motion-patterns.py Wed Jul 05 13:16:47 2017 -0400 @@ -67,7 +67,8 @@ clusterSizes = ml.computeClusterSizes(labels, prototypeIndices, -1) print(clusterSizes) -storage.savePrototypesToSqlite(args.databaseFilename, [objects[i].getNum() for i in prototypeIndices], prototypeType, [clusterSizes[i] for i in prototypeIndices]) # if saving filenames, add for example [objects[i].dbFilename for i in prototypeIndices] +prototypes = [objects[i] for i in prototypeIndices] +storage.savePrototypesToSqlite(args.databaseFilename, [p.getNum() for p in prototypes], prototypeType, prototypes, [clusterSizes[i] for i in prototypeIndices]) # if saving filenames, add for example [objects[i].dbFilename for i in prototypeIndices] if args.saveSimilarities: np.savetxt(utils.removeExtension(args.databaseFilename)+'-prototype-similarities.txt.gz', similarities, '%.4f')