From aa5996d58e68edfbefe51061856aecd549dd09c4 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 13 Jan 2015 01:27:08 +0000
Subject: [PATCH] Faster
---
src/data.c | 5 ++++-
1 files changed, 4 insertions(+), 1 deletions(-)
diff --git a/src/data.c b/src/data.c
index 12dc101..ec9f9a7 100644
--- a/src/data.c
+++ b/src/data.c
@@ -72,11 +72,14 @@
{
int i;
memset(truth, 0, k*sizeof(float));
+ int count = 0;
for(i = 0; i < k; ++i){
if(strstr(path, labels[i])){
truth[i] = 1;
+ ++count;
}
}
+ if(count != 1) printf("%d, %s\n", count, path);
}
matrix load_image_paths(char **paths, int n, int h, int w)
@@ -111,7 +114,7 @@
{
matrix y = make_matrix(n, k);
int i;
- for(i = 0; i < n; ++i){
+ for(i = 0; i < n && labels; ++i){
fill_truth(paths[i], labels, k, y.vals[i]);
}
return y;
--
Gitblit v1.10.0