From 5239186d721ac46d57198133d5aafa85fef6d22b Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Thu, 26 Nov 2015 21:46:44 +0000
Subject: [PATCH] stuff
---
src/classifier.c | 2 +-
src/yolo_kernels.cu | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/classifier.c b/src/classifier.c
index e243965..2f9a91f 100644
--- a/src/classifier.c
+++ b/src/classifier.c
@@ -279,7 +279,7 @@
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){
diff --git a/src/yolo_kernels.cu b/src/yolo_kernels.cu
index a8a131c..26487a7 100644
--- a/src/yolo_kernels.cu
+++ b/src/yolo_kernels.cu
@@ -60,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;
}
--
Gitblit v1.10.0