Fix detection paths

pull/754/head
Philip Kahn 7 years ago committed by GitHub
parent 29f3c37314
commit 3a2a8653a6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 6
      darknet.py

@ -103,8 +103,8 @@ if os.name == "nt":
raise ValueError("ForceCPU") raise ValueError("ForceCPU")
except NameError: except NameError:
pass pass
# print(os.environ.keys()) #print(os.environ.keys())
print("FORCE_CPU flag undefined, proceeding with GPU") #print("FORCE_CPU flag undefined, proceeding with GPU")
if not os.path.exists(winGPUdll): if not os.path.exists(winGPUdll):
raise ValueError("NoDLL") raise ValueError("NoDLL")
lib = CDLL(winGPUdll, RTLD_GLOBAL) lib = CDLL(winGPUdll, RTLD_GLOBAL)
@ -253,7 +253,7 @@ netMain = None
metaMain = None metaMain = None
altNames = None altNames = None
def performDetect(imagePath="data/dog.jpg", thresh= 0.25, configPath = "./yolov3.cfg", weightPath = "yolov3.weights", metaPath= "./data/coco.data", showImage= True, makeImageOnly = False, initOnly= False): def performDetect(imagePath="data/dog.jpg", thresh= 0.25, configPath = "./cfg/yolov3.cfg", weightPath = "yolov3.weights", metaPath= "./data/coco.data", showImage= True, makeImageOnly = False, initOnly= False):
""" """
Convenience function to handle the detection and returns of objects. Convenience function to handle the detection and returns of objects.

Loading…
Cancel
Save