From 56b6561ae4f1e238ba1a65701f91b40636037cc2 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 24 Mar 2015 20:20:56 +0000
Subject: [PATCH] stuff changed probably
---
src/data.c | 12 +++++++-----
1 files changed, 7 insertions(+), 5 deletions(-)
diff --git a/src/data.c b/src/data.c
index 8dd7d9a..7069248 100644
--- a/src/data.c
+++ b/src/data.c
@@ -137,18 +137,20 @@
if(j < 0) j = 0;
if(j >= num_height) j = num_height-1;
- float dw = (x - i*box_width)/box_width;
- float dh = (y - j*box_height)/box_height;
+ float dw = constrain(0,1, (x - i*box_width)/box_width );
+ float dh = constrain(0,1, (y - j*box_height)/box_height );
+ float th = constrain(0,1, h*(height+jitter)/height );
+ float tw = constrain(0,1, w*(width+jitter)/width );
int index = (i+j*num_width)*(4+classes+background);
- if(truth[index+classes+background]) continue;
+ if(truth[index+classes+background+2]) continue;
if(background) truth[index++] = 0;
truth[index+id] = 1;
index += classes;
truth[index++] = dh;
truth[index++] = dw;
- truth[index++] = h*(height+jitter)/height;
- truth[index++] = w*(width+jitter)/width;
+ truth[index++] = th;
+ truth[index++] = tw;
}
free(boxes);
}
--
Gitblit v1.10.0