From b5936b499abc94c0efffbcc99b5698574b59d860 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Sat, 05 Sep 2015 00:52:44 +0000
Subject: [PATCH] lots of stuff
---
src/detection_layer.c | 13 ++++---------
1 files changed, 4 insertions(+), 9 deletions(-)
diff --git a/src/detection_layer.c b/src/detection_layer.c
index 6a25819..80b606b 100644
--- a/src/detection_layer.c
+++ b/src/detection_layer.c
@@ -39,8 +39,8 @@
l.output = calloc(batch*outputs, sizeof(float));
l.delta = calloc(batch*outputs, sizeof(float));
#ifdef GPU
- l.output_gpu = cuda_make_array(0, batch*outputs);
- l.delta_gpu = cuda_make_array(0, batch*outputs);
+ l.output_gpu = cuda_make_array(l.output, batch*outputs);
+ l.delta_gpu = cuda_make_array(l.delta, batch*outputs);
#endif
fprintf(stderr, "Detection Layer\n");
@@ -97,6 +97,7 @@
truth.y = state.truth[j+1]/7;
truth.w = pow(state.truth[j+2], 2);
truth.h = pow(state.truth[j+3], 2);
+
box out;
out.x = l.output[j+0]/7;
out.y = l.output[j+1]/7;
@@ -107,13 +108,6 @@
float iou = box_iou(out, truth);
avg_iou += iou;
++count;
- dbox delta = diou(out, truth);
-
- l.delta[j+0] = 10 * delta.dx/7;
- l.delta[j+1] = 10 * delta.dy/7;
- l.delta[j+2] = 10 * delta.dw * 2 * sqrt(out.w);
- l.delta[j+3] = 10 * delta.dh * 2 * sqrt(out.h);
-
*(l.cost) += pow((1-iou), 2);
l.delta[j+0] = 4 * (state.truth[j+0] - l.output[j+0]);
@@ -204,6 +198,7 @@
backward_detection_layer(l, cpu_state);
cuda_push_array(state.delta, delta_cpu, l.batch*l.inputs);
+ if (truth_cpu) free(truth_cpu);
free(in_cpu);
free(delta_cpu);
}
--
Gitblit v1.10.0