diff --git a/darknet.py b/darknet.py index 4e56d98f..75ef04b5 100644 --- 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)