src/yolo_kernels.cu
@@ -21,6 +21,7 @@ extern "C" void draw_yolo(image im, int num, float thresh, box *boxes, float **probs); extern "C" char *voc_names[]; extern "C" image voc_labels[]; static float **probs; static box *boxes; @@ -59,7 +60,7 @@ printf("\033[1;1H"); printf("\nFPS:%.0f\n",fps); printf("Objects:\n\n"); draw_detections(det, l.side*l.side*l.n, thresh, boxes, probs, voc_names, 0, 20); draw_detections(det, l.side*l.side*l.n, thresh, boxes, probs, voc_names, voc_labels, 20); return 0; }