Fixed windows paths for vision

amymcgovern · Feb 11, 2018 · 7fa90cc · 7fa90cc
1 parent 1611c68
commit 7fa90cc
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/DroneVision.py b/DroneVision.py
@@ -77,9 +77,15 @@ def open_video(self):
         # we have bypassed the old opencv VideoCapture method because it was unreliable for rtsp
         fullPath = inspect.getfile(DroneVision)
         shortPathIndex = fullPath.rfind("/")
+        if (shortPathIndex == -1):
+            # handle Windows paths
+            shortPathIndex = fullPath.rfind("\\")
+        print(shortPathIndex)
         shortPath = fullPath[0:shortPathIndex]
         imagePath = join(shortPath, "images")
         utilPath = join(shortPath, "utils")
+        print(imagePath)
+        print(utilPath)
 
         # the first step is to open the rtsp stream through ffmpeg first
         # this step creates a directory full of images, one per frame