From 2313a8eb54d703323279c0fb9b2c9c52d26f0cf9 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Fri, 06 Mar 2015 18:49:03 +0000
Subject: [PATCH] Split commands into different files
---
src/data.c | 12 +++++++++---
1 files changed, 9 insertions(+), 3 deletions(-)
diff --git a/src/data.c b/src/data.c
index a6b6db3..46fba4b 100644
--- a/src/data.c
+++ b/src/data.c
@@ -89,8 +89,7 @@
float dw = (x - i*box_width)/box_width;
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+1);
- truth[index++] = 1;
+ int index = (i+j*num_width)*(4+classes);
truth[index+id] = 1;
index += classes;
truth[index++] = dh;
@@ -98,6 +97,12 @@
truth[index++] = h*(height+jitter)/height;
truth[index++] = w*(width+jitter)/width;
}
+ int i, j;
+ for(i = 0; i < num_height*num_width*(4+classes); i += 4+classes){
+ int background = 1;
+ for(j = i; j < i+classes; ++j) if (truth[j]) background = 0;
+ truth[i+classes-1] = background;
+ }
fclose(file);
}
@@ -209,7 +214,7 @@
data d;
d.shallow = 0;
d.X = load_image_paths(random_paths, n, h, w);
- int k = nh*nw*(4+classes+1);
+ int k = nh*nw*(4+classes);
d.y = make_matrix(n, k);
for(i = 0; i < n; ++i){
int dx = rand()%jitter;
@@ -227,6 +232,7 @@
void *load_detection_thread(void *ptr)
{
+ printf("Loading data: %d\n", rand());
struct load_args a = *(struct load_args*)ptr;
*a.d = load_data_detection_jitter_random(a.n, a.paths, a.m, a.classes, a.h, a.w, a.nh, a.nw, a.jitter);
translate_data_rows(*a.d, -128);
--
Gitblit v1.10.0