From 59e356730fa62d1e307d4fc7a5657f8492633088 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 22 Sep 2015 22:24:54 +0000
Subject: [PATCH] writing stuff
---
src/region_layer.c | 4 +++-
1 files changed, 3 insertions(+), 1 deletions(-)
diff --git a/src/region_layer.c b/src/region_layer.c
index ecb89c6..39af5ee 100644
--- a/src/region_layer.c
+++ b/src/region_layer.c
@@ -61,6 +61,7 @@
if(state.train){
float avg_iou = 0;
float avg_cat = 0;
+ float avg_allcat = 0;
float avg_obj = 0;
float avg_anyobj = 0;
int count = 0;
@@ -90,6 +91,7 @@
l.delta[class_index+j] = l.class_scale * (state.truth[truth_index+1+j] - l.output[class_index+j]);
*(l.cost) += l.class_scale * pow(state.truth[truth_index+1+j] - l.output[class_index+j], 2);
if(state.truth[truth_index + 1 + j]) avg_cat += l.output[class_index+j];
+ avg_allcat += l.output[class_index+j];
}
box truth = float_to_box(state.truth + truth_index + 1 + l.classes);
@@ -151,7 +153,7 @@
LOGISTIC, l.delta + index + locations*l.classes);
}
}
- printf("Region Avg IOU: %f, Avg Cat Pred: %f, Avg Obj: %f, Avg Any: %f, count: %d\n", avg_iou/count, avg_cat/count, avg_obj/count, avg_anyobj/(l.batch*locations*l.n), count);
+ printf("Region Avg IOU: %f, Pos Cat: %f, All Cat: %f, Pos Obj: %f, Any Obj: %f, count: %d\n", avg_iou/count, avg_cat/count, avg_allcat/(count*l.classes), avg_obj/count, avg_anyobj/(l.batch*locations*l.n), count);
}
}
--
Gitblit v1.10.0