From d7d7da2653ff4f79a275529b0ac3fec438880083 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Fri, 27 Mar 2015 02:13:59 +0000
Subject: [PATCH] Fixed im2col mistake >< face#palm
---
src/detection.c | 9 +++++----
1 files changed, 5 insertions(+), 4 deletions(-)
diff --git a/src/detection.c b/src/detection.c
index 1800ca6..69202aa 100644
--- a/src/detection.c
+++ b/src/detection.c
@@ -45,7 +45,7 @@
{
char *base = basecfg(cfgfile);
printf("%s\n", base);
- float avg_loss = 1;
+ float avg_loss = -1;
network net = parse_network_cfg(cfgfile);
if(weightfile){
load_weights(&net, weightfile);
@@ -84,6 +84,7 @@
time=clock();
float loss = train_network(net, train);
net.seen += imgs;
+ if (avg_loss < 0) avg_loss = loss;
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==0){
@@ -109,8 +110,8 @@
char **paths = (char **)list_to_array(plist);
int im_size = 448;
int classes = 20;
- int background = 1;
- int nuisance = 0;
+ int background = 0;
+ int nuisance = 1;
int num_output = 7*7*(4+classes+background+nuisance);
int m = plist->size;
@@ -137,7 +138,7 @@
for(j = 0; j < pred.rows; ++j){
for(k = 0; k < pred.cols; k += classes+4+background+nuisance){
float scale = 1.;
- if(nuisance) scale = pred.vals[j][k];
+ if(nuisance) scale = 1.-pred.vals[j][k];
for(class = 0; class < classes; ++class){
int index = (k)/(classes+4+background+nuisance);
int r = index/7;
--
Gitblit v1.10.0