Mercurial Hosting > traffic-intelligence
comparison 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 |
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: |