| | |
|---|
| | | #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) |
|---|
| | |
|---|
| | | if debug: print("Assigned pnum") |
|---|
| | | predict_image(net, im) |
|---|
| | | if debug: print("did prediction") |
|---|
| | | dets = get_network_boxes(net, im.w, im.h, thresh, hier_thresh, None, 0, pnum, 1) |
|---|
| | | dets = get_network_boxes(net, im.w, im.h, thresh, hier_thresh, None, 0, pnum, 0) |
|---|
| | | if debug: print("Got dets") |
|---|
| | | num = pnum[0] |
|---|
| | | if debug: print("got zeroth index of pnum") |
|---|
