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

apache · Oct 18, 2018 · f1a4b94 · f1a4b94
1 parent a6f32bf
commit f1a4b94
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nnvm/python/nnvm/testing/yolo_detection.py b/nnvm/python/nnvm/testing/yolo_detection.py
@@ -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)