Mercurial Hosting > traffic-intelligence
comparison scripts/learn-motion-patterns.py @ 849:a414a7d58483
corrected issue with prototypes storage
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Fri, 22 Jul 2016 17:23:49 -0400 |
parents | 0cb69238e6f5 |
children | 1535251a1f40 |
comparison
equal
deleted
inserted
replaced
848:0cb69238e6f5 | 849:a414a7d58483 |
---|---|
51 prototypeIndices, labels = ml.prototypeCluster(trajectories, similarities, args.minSimilarity, lambda x,y : lcss.computeNormalized(x, y), args.minClusterSize, args.randomInitialization) # this line can be called again without reinitializing similarities | 51 prototypeIndices, labels = ml.prototypeCluster(trajectories, similarities, args.minSimilarity, lambda x,y : lcss.computeNormalized(x, y), args.minClusterSize, args.randomInitialization) # this line can be called again without reinitializing similarities |
52 | 52 |
53 clusterSizes = ml.computeClusterSizes(labels, prototypeIndices, -1) | 53 clusterSizes = ml.computeClusterSizes(labels, prototypeIndices, -1) |
54 print(clusterSizes) | 54 print(clusterSizes) |
55 | 55 |
56 storage.savePrototypesToSqlite(args.databaseFilename, [objects[i].getNum() for i in prototypeIndices], args.trajectoryType, {objects[i].getNum():clusterSizes[i] for i in prototypeIndices}) # if saving filenames, add for example {objects[i].getNum():objects[i].dbFilename for i in prototypeIndices} | 56 storage.savePrototypesToSqlite(args.databaseFilename, [objects[i].getNum() for i in prototypeIndices], args.trajectoryType, [clusterSizes[i] for i in prototypeIndices]) # if saving filenames, add for example [objects[i].dbFilename for i in prototypeIndices] |
57 | 57 |
58 if args.saveSimilarities: | 58 if args.saveSimilarities: |
59 np.savetxt(utils.removeExtension(args.databaseFilename)+'-prototype-similarities.txt.gz', similarities, '%.4f') | 59 np.savetxt(utils.removeExtension(args.databaseFilename)+'-prototype-similarities.txt.gz', similarities, '%.4f') |
60 | 60 |
61 if args.display: | 61 if args.display: |