# HG changeset patch
# User Nicolas Saunier <nicolas.saunier@polymtl.ca>
# Date 1433503611 -7200
# Node ID 6d2ece4aed7c2383348dc67d81fb63f079c0be57
# Parent  f3a0b652b590690b5f0eef65fbe7194056ee30cf
bug forgot to svae objects

diff -r f3a0b652b590 -r 6d2ece4aed7c scripts/classify-objects.py
--- a/scripts/classify-objects.py	Fri Jun 05 11:04:06 2015 +0200
+++ b/scripts/classify-objects.py	Fri Jun 05 13:26:51 2015 +0200
@@ -72,6 +72,7 @@
 width = int(capture.get(cv2.cv.CV_CAP_PROP_FRAME_WIDTH))
 height = int(capture.get(cv2.cv.CV_CAP_PROP_FRAME_HEIGHT))
 
+pastObjects = []
 if params.undistort: # setup undistortion
     [map1, map2] = computeUndistortMaps(width, height, undistortedImageMultiplication, intrinsicCameraMatrix, distortionCoefficients)
 if capture.isOpened():
@@ -87,8 +88,9 @@
                 print('frame number: {}'.format(frameNum))
                 currentObjects = []
                 for obj in objects:
-                    if obj.getLastFrameNum() < frameNum:
+                    if obj.getLastInstant() < frameNum:
                         obj.classifyUserTypeHoGSVM(minSpeedEquiprobable = params.minSpeedEquiprobable, speedProbabilities = speedProbabilities)
+                        pastObjects.append(obj)
                     else:
                         currentObjects.append(obj)
                 objects = currentObjects
@@ -103,3 +105,6 @@
     
     for obj in objects:
         obj.classifyUserTypeHoGSVM(minSpeedEquiprobable = params.minSpeedEquiprobable, speedProbabilities = speedProbabilities)
+        pastObjects.append(obj)
+    print('Saving user types')
+    storage.setRoadUserTypes(params.databaseFilename, pastObjects)