darknet.py - loading dll files by absolute name

pull/856/head
Vinjn Zhang 7 years ago committed by GitHub
parent efdeb01a3a
commit dbcdac88a5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 6
      darknet.py

@ -78,8 +78,10 @@ class METADATA(Structure):
#lib = CDLL("darknet.so", RTLD_GLOBAL) #lib = CDLL("darknet.so", RTLD_GLOBAL)
hasGPU = True hasGPU = True
if os.name == "nt": if os.name == "nt":
winGPUdll = "yolo_cpp_dll.dll" cwd = os.path.dirname(__file__)
winNoGPUdll = "yolo_cpp_dll_nogpu.dll" 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() envKeys = list()
for k, v in os.environ.items(): for k, v in os.environ.items():
envKeys.append(k) envKeys.append(k)

Loading…
Cancel
Save