mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

darknet.py - apply the same changes to build/darknet/x64/

Browse Source
This commit is contained in:
Vinjn Zhang
2018-05-20 21:12:35 +08:00
committed by GitHub
parent dbcdac88a5
commit c6ca54b411
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
build/darknet/x64/darknet.py
View File

@ -78,13 +78,14 @@ 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)
try: try:
tmp = os.environ["CUDA_HOME"]
try: try:
tmp = os.environ["FORCE_CPU"].lower() tmp = os.environ["FORCE_CPU"].lower()
if tmp in ["1", "true", "yes", "on"]: if tmp in ["1", "true", "yes", "on"]: