Mercurial Hosting > traffic-intelligence

comparison scripts/learn-motion-patterns.py @ 919:7b3f2e0a2652

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
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
comparison
equal deleted inserted replaced
918:3a06007a4bb7 919:7b3f2e0a2652
65 prototypeIndices, labels = ml.prototypeCluster(trajectories, similarities, args.minSimilarity, lambda x,y : lcss.computeNormalized(x, y), args.minClusterSize, args.optimizeCentroid, args.randomInitialization, True, None) # this line can be called again without reinitializing similarities 65 prototypeIndices, labels = ml.prototypeCluster(trajectories, similarities, args.minSimilarity, lambda x,y : lcss.computeNormalized(x, y), args.minClusterSize, args.optimizeCentroid, args.randomInitialization, True, None) # this line can be called again without reinitializing similarities
66 66
67 clusterSizes = ml.computeClusterSizes(labels, prototypeIndices, -1) 67 clusterSizes = ml.computeClusterSizes(labels, prototypeIndices, -1)
68 print(clusterSizes) 68 print(clusterSizes)
69 69
70 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] 70 prototypes = [objects[i] for i in prototypeIndices]
71 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]
71 72
72 if args.saveSimilarities: 73 if args.saveSimilarities:
73 np.savetxt(utils.removeExtension(args.databaseFilename)+'-prototype-similarities.txt.gz', similarities, '%.4f') 74 np.savetxt(utils.removeExtension(args.databaseFilename)+'-prototype-similarities.txt.gz', similarities, '%.4f')
74 75
75 # if args.saveMatches: 76 # if args.saveMatches: