| | |
|---|
| | | #pylint: disable= C0321 |
|---|
| | | im = load_image(image, 0, 0) |
|---|
| | | #import cv2 |
|---|
| | | #custom_image = cv2.imread(image) # use: detect(,,imagePath,) |
|---|
| | | #custom_image_bgr = cv2.imread(image) # use: detect(,,imagePath,) |
|---|
| | | #custom_image = cv2.cvtColor(custom_image_bgr, cv2.COLOR_BGR2RGB) |
|---|
| | | #custom_image = cv2.resize(custom_image,(lib.network_width(net), lib.network_height(net)), interpolation = cv2.INTER_LINEAR) |
|---|
| | | #import scipy.misc |
|---|
| | | #custom_image = scipy.misc.imread(image) |
|---|
| | | #im, arr = array_to_image(custom_image) # you should comment line below: free_image(im) |
|---|
| | |
|---|
| | | 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, custom_image_bgr.shape[1], custom_image_bgr.shape[0], thresh, hier_thresh, None, 0, pnum, 0) # OpenCV |
|---|
| | | 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") |
|---|
| | |
|---|
| | | if not os.path.exists(imagePath): |
|---|
| | | raise ValueError("Invalid image path `"+os.path.abspath(imagePath)+"`") |
|---|
| | | # Do the detection |
|---|
| | | #detections = detect(netMain, metaMain, imagePath, thresh) # if is used cv2.imread(image) |
|---|
| | | detections = detect(netMain, metaMain, imagePath.encode("ascii"), thresh) |
|---|
| | | if showImage: |
|---|
| | | try: |
|---|
