diff scripts/learn-motion-patterns.py @ 920:499154254f37

improved prototype loading
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Wed, 05 Jul 2017 16:30:04 -0400
parents 7b3f2e0a2652
children 630934595871
line wrap: on
line diff
--- a/scripts/learn-motion-patterns.py	Wed Jul 05 13:16:47 2017 -0400
+++ b/scripts/learn-motion-patterns.py	Wed Jul 05 16:30:04 2017 -0400
@@ -10,8 +10,9 @@
 parser = argparse.ArgumentParser(description='The program learns prototypes for the motion patterns') #, epilog = ''
 #parser.add_argument('--cfg', dest = 'configFilename', help = 'name of the configuration file')
 parser.add_argument('-d', dest = 'databaseFilename', help = 'name of the Sqlite database file', required = True)
+parser.add_argument('-r', dest = 'initialPrototypeDatabaseFilename', help = 'name of the Sqlite database file for prototypes to start the algorithm with')
 parser.add_argument('-t', dest = 'trajectoryType', help = 'type of trajectories to learn from', choices = ['objectfeatures', 'feature', 'object'], default = 'objectfeatures')
-parser.add_argument('--max-nobjectfeatures', dest = 'maxNObjectFeatures', help = 'maximum number of features per object to load', type = int, default = 3)
+parser.add_argument('--max-nobjectfeatures', dest = 'maxNObjectFeatures', help = 'maximum number of features per object to load', type = int, default = 1)
 parser.add_argument('-n', dest = 'nTrajectories', help = 'number of the object or feature trajectories to load', type = int, default = None)
 parser.add_argument('-e', dest = 'epsilon', help = 'distance for the similarity of trajectory points', type = float, required = True)
 parser.add_argument('--metric', dest = 'metric', help = 'metric for the similarity of trajectory points', default = 'cityblock') # default is manhattan distance
@@ -30,13 +31,11 @@
 # 1. learn proto from one file, save in same or another (with traj)
 # 2. load proto, load objects, update proto, save proto
 # 3. assign objects from one db to proto
-# 4. load objects from several files, save in another
+# 4. load objects from several files, save in another -> see metadata: site with view and times
 # 5. keep prototypes, with positions/velocities, in separate db (keep link to original data through filename, type and index)
 
 # TODO add possibility to cluter with velocities
 # TODO add possibility to start with saved prototypes so that one can incrementally learn from several databases
-# save prototypes with database name, add option to keep trajectory along: if saved in same db, no need
-# load proto must load the movingobject
 # save the objects that match the prototypes
 # write an assignment function for objects
 
@@ -68,7 +67,7 @@
 print(clusterSizes)
 
 prototypes = [objects[i] for i in prototypeIndices]
-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]
+storage.savePrototypesToSqlite(args.databaseFilename, [p.getNum() for p in prototypes], prototypeType, [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')