Mercurial Hosting > traffic-intelligence
diff 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 |
line wrap: on
line diff
--- a/scripts/learn-motion-patterns.py Thu Jul 21 17:51:11 2016 -0400 +++ b/scripts/learn-motion-patterns.py Fri Jul 22 17:23:49 2016 -0400 @@ -53,7 +53,7 @@ clusterSizes = ml.computeClusterSizes(labels, prototypeIndices, -1) print(clusterSizes) -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} +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] if args.saveSimilarities: np.savetxt(utils.removeExtension(args.databaseFilename)+'-prototype-similarities.txt.gz', similarities, '%.4f')