mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

complete get_avgpool_image

Browse Source 
get_avgpool_image is declared in avgpool_layer.h while it is not defined in avgpool_layer.c.
This commit is contained in:
DoctorKey 2018-04-13 09:40:22 +08:00
parent f61f3171d9
commit 817106e531
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/avgpool_layer.c
View File

@ -2,6 +2,14 @@
#include "cuda.h" #include "cuda.h"
#include <stdio.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) 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); fprintf(stderr, "avg %4d x%4d x%4d -> %4d\n", w, h, c, c);