stuff

src/classifier.c

@@ -279,7 +279,7 @@ void test_classifier(char *datacfg, char *cfgfile, char *weightfile, char *filen
         
         int i;
         if (target_layer >= 0){
-            //layer l = net.layers[target_layer];
+            layer l = net.layers[target_layer];
         }
 
         for(i = 0; i < val.X.rows; ++i){

src/yolo_kernels.cu

@@ -60,7 +60,7 @@ void *detect_in_thread(void *ptr)
     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;
 }