darknet.py - loading dll files by absolute name

7 years ago · dbcdac88a5
parent efdeb01a3a
commit dbcdac88a5
1 changed files with 4 additions and 2 deletions
--- a/darknet.py
+++ b/darknet.py
@ -78,8 +78,10 @@ class METADATA(Structure):
 #lib = CDLL("darknet.so", RTLD_GLOBAL)
 hasGPU = True
 if os.name == "nt":
-    winGPUdll = "yolo_cpp_dll.dll"
-    winNoGPUdll = "yolo_cpp_dll_nogpu.dll"
+    cwd = os.path.dirname(__file__)
+    os.environ['PATH'] = cwd + ';' + os.environ['PATH']
+    winGPUdll = os.path.join(cwd, "yolo_cpp_dll.dll")
+    winNoGPUdll = os.path.join(cwd, "yolo_cpp_dll_nogpu.dll")
    envKeys = list()
    for k, v in os.environ.items():
        envKeys.append(k)