Use current working directory to store files instead of specific directory

Author: javanese-programmer

collect.py

@@ -1,10 +1,11 @@
 """Scripts to collect images without running a detection."""
 
 import sys
+import os
 
 import cv2
 
-IMG_ADDR = "/home/pi/Desktop/Object Detection/Raspberry Pi"
+IMG_ADDR = str(os.getcwd())
 
 counter = 0
 

test/deeplearning_test.py

@@ -14,7 +14,7 @@
 from utils import vizres, array, plot
 
 # Add project directory to path before importing modules
-sys.path.insert(0, "/home/pi/Desktop/Object Detection/Raspberry Pi")
+sys.path.insert(0, str(os.getcwd()))
 
 
 def parse_argument():
@@ -32,7 +32,7 @@ def parse_argument():
         "--imgPath",
         help="Path of the test images.",
         required=False,
-        default="/home/pi/Desktop/Object Detection/Raspberry Pi/test_data",
+        default=str(os.getcwd()) + "/test_data",
     )
     parser.add_argument(
         "--numThreads",

test/traditional_test.py

@@ -11,7 +11,7 @@
 from utils import vizres, array, plot
 
 # Add project directory to path before importing modules
-sys.path.insert(0, "/home/pi/Desktop/Object Detection/Raspberry Pi")
+sys.path.insert(0, str(os.getcwd()))
 
 
 def parse_argument():
@@ -30,7 +30,7 @@ def parse_argument():
         "--imgPath",
         help="Path of the test images.",
         required=False,
-        default="/home/pi/Desktop/Object Detection/Raspberry Pi/test_data",
+        default=str(os.getcwd()) + "/test_data",
     )
     parser.add_argument(
         "--csvFilename",

utils/video.py

@@ -9,9 +9,9 @@
 
 
 # Define images and parent directory
-IMAGE_DIR = r"/home/pi/Desktop/Object Detection/Raspberry Pi/image"
-VIDEO_DIR = r"/home/pi/Desktop/Object Detection/Raspberry Pi/video"
-PARENT_DIR = r"/home/pi/Desktop/Object Detection/Raspberry Pi"
+PARENT_DIR = str(os.getcwd())
+IMAGE_DIR = PARENT_DIR + '/' + 'image'
+VIDEO_DIR = PARENT_DIR + '/' + 'video'
 
 
 def save_img(