changeset 835:f3ae72d86762

added option to subsample positions to learn motion patterns
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Tue, 05 Jul 2016 13:56:13 -0400
parents 119c4efe6398
children 7058a40a4bbc
files scripts/learn-motion-patterns.py
diffstat 1 files changed, 2 insertions(+), 1 deletions(-) [+]
line wrap: on
line diff
--- a/scripts/learn-motion-patterns.py	Thu Jun 30 14:19:59 2016 -0400
+++ b/scripts/learn-motion-patterns.py	Tue Jul 05 13:56:13 2016 -0400
@@ -17,6 +17,7 @@
 parser.add_argument('--metric', dest = 'metric', help = 'metric for the similarity of trajectory points', default = 'cityblock') # default is manhattan distance
 parser.add_argument('-s', dest = 'minSimilarity', help = 'minimum similarity to put a trajectory in a cluster', type = float, required = True)
 parser.add_argument('-c', dest = 'minClusterSize', help = 'minimum cluster size', type = int, default = None)
+parser.add_argument('--subsample', dest = 'positionSubsamplingRate', help = 'rate of position subsampling (1 every n positions)', type = int, default = None)
 parser.add_argument('--display', dest = 'display', help = 'display trajectories', action = 'store_true') # default is manhattan distance
 
 args = parser.parse_args()
@@ -29,7 +30,7 @@
     trajectoryType = 'object'
 
 #features = storage.loadTrajectoriesFromSqlite(databaseFilename, args.trajectoryType)
-objects = storage.loadTrajectoriesFromSqlite(args.databaseFilename, trajectoryType, withFeatures = (args.trajectoryType == 'objectfeatures'), objectNumbers = args.nTrajectories)
+objects = storage.loadTrajectoriesFromSqlite(args.databaseFilename, trajectoryType, withFeatures = (args.trajectoryType == 'objectfeatures'), objectNumbers = args.nTrajectories, timeStep = args.positionSubsamplingRate)
 
 if args.trajectoryType == 'objectfeatures':
     features = []