From 655f636a42d6e1d4518b712cfac6d973424de693 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Sun, 08 Mar 2015 18:25:28 +0000
Subject: [PATCH] detection layer fixed
---
src/data.c | 11 +++++++++++
1 files changed, 11 insertions(+), 0 deletions(-)
diff --git a/src/data.c b/src/data.c
index 46fba4b..a429476 100644
--- a/src/data.c
+++ b/src/data.c
@@ -90,6 +90,7 @@
float dh = (y - j*box_height)/box_height;
//printf("%d %d %d %f %f\n", id, i, j, dh, dw);
int index = (i+j*num_width)*(4+classes);
+ if(truth[index+classes]) continue;
truth[index+id] = 1;
index += classes;
truth[index++] = dh;
@@ -148,6 +149,16 @@
return d;
}
+data load_data_captcha_encode(char **paths, int n, int m, int h, int w)
+{
+ if(m) paths = get_random_paths(paths, n, m);
+ data d;
+ d.shallow = 0;
+ d.X = load_image_paths(paths, n, h, w);
+ d.y = d.X;
+ if(m) free(paths);
+ return d;
+}
void fill_truth(char *path, char **labels, int k, float *truth)
{
--
Gitblit v1.10.0