From 956cfcaec993111426d91bcd61676b5fe0ebfd16 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Mon, 24 Feb 2014 21:02:53 +0000
Subject: [PATCH] Feature extraction using Imagenet
---
src/softmax_layer.c | 5 ++++-
1 files changed, 4 insertions(+), 1 deletions(-)
diff --git a/src/softmax_layer.c b/src/softmax_layer.c
index 1e01bd2..b6b7ff3 100644
--- a/src/softmax_layer.c
+++ b/src/softmax_layer.c
@@ -36,8 +36,11 @@
}
for(i = 0; i < layer.inputs; ++i){
sum += exp(input[i]-largest);
+ //printf("%f, ", input[i]);
}
- sum = largest+log(sum);
+ //printf("\n");
+ if(sum) sum = largest+log(sum);
+ else sum = largest-100;
for(i = 0; i < layer.inputs; ++i){
layer.output[i] = exp(input[i]-sum);
}
--
Gitblit v1.10.0