Skip to content
Snippets Groups Projects
Commit f1a4b94b authored by Siju's avatar Siju Committed by Tianqi Chen
Browse files

[YOLO]Add the probability to the image (#1910)

parent a6f32bff
No related branches found
No related tags found
No related merge requests found
......@@ -160,7 +160,7 @@ def draw_detections(im, dets, thresh, names, classes):
if det['prob'][j] > thresh:
if category == -1:
category = j
labelstr.append(names[j])
labelstr.append(names[j] + " " + str(round(det['prob'][j], 4)))
if category > -1:
imc, imh, imw = im.shape
width = int(imh * 0.006)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment