From 11c72b1132feca7c1252ea01d02da4cb497e723f Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Thu, 11 Jun 2015 22:38:58 +0000
Subject: [PATCH] testing on one image

---
 src/activations.c |   14 ++++++++++++++
 1 files changed, 14 insertions(+), 0 deletions(-)

diff --git a/src/activations.c b/src/activations.c
index 7da5ce2..8b607c2 100644
--- a/src/activations.c
+++ b/src/activations.c
@@ -12,12 +12,16 @@
             return "logistic";
         case RELU:
             return "relu";
+        case RELIE:
+            return "relie";
         case RAMP:
             return "ramp";
         case LINEAR:
             return "linear";
         case TANH:
             return "tanh";
+        case PLSE:
+            return "plse";
         default:
             break;
     }
@@ -28,6 +32,8 @@
 {
     if (strcmp(s, "logistic")==0) return LOGISTIC;
     if (strcmp(s, "relu")==0) return RELU;
+    if (strcmp(s, "relie")==0) return RELIE;
+    if (strcmp(s, "plse")==0) return PLSE;
     if (strcmp(s, "linear")==0) return LINEAR;
     if (strcmp(s, "ramp")==0) return RAMP;
     if (strcmp(s, "tanh")==0) return TANH;
@@ -44,10 +50,14 @@
             return logistic_activate(x);
         case RELU:
             return relu_activate(x);
+        case RELIE:
+            return relie_activate(x);
         case RAMP:
             return ramp_activate(x);
         case TANH:
             return tanh_activate(x);
+        case PLSE:
+            return plse_activate(x);
     }
     return 0;
 }
@@ -69,10 +79,14 @@
             return logistic_gradient(x);
         case RELU:
             return relu_gradient(x);
+        case RELIE:
+            return relie_gradient(x);
         case RAMP:
             return ramp_gradient(x);
         case TANH:
             return tanh_gradient(x);
+        case PLSE:
+            return plse_gradient(x);
     }
     return 0;
 }

--
Gitblit v1.10.0