From 8561e49b5a2876e9a522b2dedfa99f19d5738154 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Mon, 13 Jul 2015 22:04:21 +0000
Subject: [PATCH] add avgpool layer
---
src/network.c | 10 ++++++++++
1 files changed, 10 insertions(+), 0 deletions(-)
diff --git a/src/network.c b/src/network.c
index 53608e6..e85dfe9 100644
--- a/src/network.c
+++ b/src/network.c
@@ -12,6 +12,7 @@
#include "detection_layer.h"
#include "normalization_layer.h"
#include "maxpool_layer.h"
+#include "avgpool_layer.h"
#include "cost_layer.h"
#include "softmax_layer.h"
#include "dropout_layer.h"
@@ -28,6 +29,8 @@
return "connected";
case MAXPOOL:
return "maxpool";
+ case AVGPOOL:
+ return "avgpool";
case SOFTMAX:
return "softmax";
case DETECTION:
@@ -83,6 +86,8 @@
forward_softmax_layer(l, state);
} else if(l.type == MAXPOOL){
forward_maxpool_layer(l, state);
+ } else if(l.type == AVGPOOL){
+ forward_avgpool_layer(l, state);
} else if(l.type == DROPOUT){
forward_dropout_layer(l, state);
} else if(l.type == ROUTE){
@@ -156,6 +161,8 @@
backward_normalization_layer(l, state);
} else if(l.type == MAXPOOL){
if(i != 0) backward_maxpool_layer(l, state);
+ } else if(l.type == AVGPOOL){
+ backward_avgpool_layer(l, state);
} else if(l.type == DROPOUT){
backward_dropout_layer(l, state);
} else if(l.type == DETECTION){
@@ -273,6 +280,9 @@
resize_convolutional_layer(&l, w, h);
}else if(l.type == MAXPOOL){
resize_maxpool_layer(&l, w, h);
+ }else if(l.type == AVGPOOL){
+ resize_avgpool_layer(&l, w, h);
+ break;
}else if(l.type == NORMALIZATION){
resize_normalization_layer(&l, w, h);
}else{
--
Gitblit v1.10.0