darknet.py - loading dll files by absolute name

This commit is contained in:
Vinjn Zhang
2018-05-20 14:20:55 +08:00
committed by GitHub
parent efdeb01a3a
commit dbcdac88a5

View File

@ -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)