From 817106e5310d1359f8b5ec7ddc0011d892676c33 Mon Sep 17 00:00:00 2001
From: DoctorKey <wangguohua_key@163.com>
Date: Fri, 13 Apr 2018 09:40:22 +0800
Subject: [PATCH] complete get_avgpool_image

get_avgpool_image is declared in avgpool_layer.h while it is not defined in avgpool_layer.c.
---
 src/avgpool_layer.c | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/src/avgpool_layer.c b/src/avgpool_layer.c
index 83034dbe..6ed74f87 100644
--- a/src/avgpool_layer.c
+++ b/src/avgpool_layer.c
@@ -2,6 +2,14 @@
 #include "cuda.h"
 #include <stdio.h>
 
+image get_avgpool_image(avgpool_layer l)
+{
+    int h = l.out_h;
+    int w = l.out_w;
+    int c = l.c;
+    return float_to_image(w, h, c, l.output);
+}
+
 avgpool_layer make_avgpool_layer(int batch, int w, int h, int c)
 {
     fprintf(stderr, "avg                     %4d x%4d x%4d   ->  %4d\n",  w, h, c, c);