diff python/storage.py @ 589:5800a87f11ae

corrected one bug and changed attribute names
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Fri, 05 Dec 2014 15:14:44 -0500
parents c5406edbcf12
children dc2d0a0d7fe1
line wrap: on
line diff
--- a/python/storage.py	Fri Dec 05 00:54:38 2014 -0500
+++ b/python/storage.py	Fri Dec 05 15:14:44 2014 -0500
@@ -286,7 +286,7 @@
                 corner = 'top_left'
             elif trajectoryType == 'bbbottom':
                 corner = 'bottom_right'
-            queryStatement = 'SELECT object_id, frame_number, x_'+corner+', y_'+corner+' FROM '+tableName+' '+trajectoryIdQuery+'ORDER BY object_id, frame_number'
+            queryStatement = 'SELECT object_id, frame_number, x_'+corner+', y_'+corner+' FROM '+tableName+' '+idQuery+'ORDER BY object_id, frame_number'
             cursor.execute(queryStatement)
             logging.debug(queryStatement)
         else:
@@ -375,7 +375,7 @@
     connection.close()
     return objects
 
-def loadGroundTruthFromSqlite(filename, gtType, gtNumbers = None):
+def loadGroundTruthFromSqlite(filename, gtType = 'bb', gtNumbers = None):
     'Loads bounding box annotations (ground truth) from an SQLite '
     connection = sqlite3.connect(filename)
     gt = []