From c521f87c9eb3ee70258f315ad0d72d7cf174e8b7 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Mon, 25 May 2015 18:53:10 +0000
Subject: [PATCH] ?
---
src/detection.c | 8 +++++++-
1 files changed, 7 insertions(+), 1 deletions(-)
diff --git a/src/detection.c b/src/detection.c
index a1ba888..c012848 100644
--- a/src/detection.c
+++ b/src/detection.c
@@ -47,6 +47,8 @@
int top = (y-h/2)*im.h;
int bot = (y+h/2)*im.h;
draw_box(im, left, top, right, bot, red, green, blue);
+ draw_box(im, left+1, top+1, right+1, bot+1, red, green, blue);
+ draw_box(im, left-1, top-1, right-1, bot-1, red, green, blue);
}
}
}
@@ -116,7 +118,11 @@
float loss = train_network(net, train);
//TODO
+ #ifdef GPU
float *out = get_network_output_gpu(net);
+ #else
+ float *out = get_network_output(net);
+ #endif
image im = float_to_image(net.w, net.h, 3, train.X.vals[127]);
image copy = copy_image(im);
draw_localization(copy, &(out[63*80]));
@@ -213,7 +219,7 @@
avg_loss = avg_loss*.9 + loss*.1;
printf("%d: %f, %f avg, %lf seconds, %d images\n", i, loss, avg_loss, sec(clock()-time), i*imgs);
if(i == 100){
- net.learning_rate *= 10;
+ //net.learning_rate *= 10;
}
if(i%100==0){
char buff[256];
--
Gitblit v1.10.0