changeset 1102:cdf1773ba89e

corrected bug
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Wed, 27 Feb 2019 16:28:08 -0500
parents e23828659a66
children 7594802f281a
files scripts/init-tracking.py
diffstat 1 files changed, 6 insertions(+), 6 deletions(-) [+]
line wrap: on
line diff
--- a/scripts/init-tracking.py	Fri Feb 22 17:22:04 2019 -0500
+++ b/scripts/init-tracking.py	Wed Feb 27 16:28:08 2019 -0500
@@ -3,7 +3,7 @@
 import sys, argparse, os.path
 from shutil import copy
 
-from cvutils import getImagesFromVideo
+from trafficintelligence.cvutils import getImagesFromVideo
 from matplotlib.pyplot import imsave
 
 from trafficintelligence import storage, utils
@@ -29,9 +29,9 @@
     directoryName = matchingPaths[0]
     if directoryName.endswith('/'):
         directoryName = directoryName[:-1]
-    if os.path.exists(directoryName+'/../tracking.cfg') and not os.path.exists('./tracking.cfg'):
-        f = storage.openCheck(directoryName+'/../tracking.cfg')
-        out = storage.openCheck('./tracking.cfg', 'w')
+    if os.path.exists(directoryName+'/tracking.cfg') and not os.path.exists('./tracking.cfg'):
+        f = utils.openCheck(directoryName+'/tracking.cfg')
+        out = utils.openCheck('./tracking.cfg', 'w')
         for l in f:
             if 'video-filename' in l:
                 tmp = l.split('=')
@@ -44,8 +44,8 @@
         f.close()
         out.close()
         print('Configuration file tracking.cfg successfully copied to the current directory with video and database filename adapted')
-    if os.path.exists(directoryName+'/../classifier.cfg') and not os.path.exists('./classifier.cfg'):
-        copy(directoryName+'/../classifier.cfg', 'classifier.cfg')
+    if os.path.exists(directoryName+'/classifier.cfg') and not os.path.exists('./classifier.cfg'):
+        copy(directoryName+'/classifier.cfg', 'classifier.cfg')
         print('Configuration file classifier.cfg successfully copied to the current directory')
         
 # extract image from video