diff python/storage.py @ 862:2d6249fe905a

correcting bug
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Mon, 07 Nov 2016 11:11:10 -0500
parents f9c9457b60c2
children 6db83beb5350
line wrap: on
line diff
--- a/python/storage.py	Fri Nov 04 11:47:42 2016 -0400
+++ b/python/storage.py	Mon Nov 07 11:11:10 2016 -0500
@@ -1258,20 +1258,23 @@
         config = ConfigParser()
         config.readfp(FakeSecHead(openCheck(filename)))
         # check if path contains directory names
-        dirname = path.split(filename)[0]
+        if len(path.split(filename)[0]) == 0:
+            dirname = '.'+path.sep
+        else:
+            dirname = path.split(filename)[0]+path.sep
 
         self.sectionHeader = config.sections()[0]
         # Tracking/display parameters
         self.videoFilename = config.get(self.sectionHeader, 'video-filename')
         self.databaseFilename = config.get(self.sectionHeader, 'database-filename')
         self.homographyFilename = config.get(self.sectionHeader, 'homography-filename')
-        if (path.exists(self.homographyFilename)):
-            self.homography = loadtxt(self.homographyFilename)
+        if path.exists(dirname+self.homographyFilename):
+            self.homography = loadtxt(dirname+self.homographyFilename)
         else:
             self.homography = None
         self.intrinsicCameraFilename = config.get(self.sectionHeader, 'intrinsic-camera-filename')
-        if (path.exists(dirname+path.sep+self.intrinsicCameraFilename)):
-            self.intrinsicCameraMatrix = loadtxt(dirname+path.sep+self.intrinsicCameraFilename)
+        if path.exists(dirname+self.intrinsicCameraFilename):
+            self.intrinsicCameraMatrix = loadtxt(dirname+self.intrinsicCameraFilename)
         else:
             self.intrinsicCameraMatrix = None
         distortionCoefficients = getValuesFromINIFile(filename, 'distortion-coefficients', '=')