Joseph Redmon
2015-11-26 5239186d721ac46d57198133d5aafa85fef6d22b
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;
}