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

maxpool fixed, good on mnist

Browse Source
This commit is contained in:
Joseph Redmon
2014-08-09 08:16:37 -07:00
parent d9f1b0b16e
commit 7add111509
5 changed files with 54 additions and 93 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/maxpool_layer.h
View File

@ -8,6 +8,7 @@ typedef struct {
int h,w,c;
int stride;
int size;
int *max_indexes;
float *delta;
float *output;
} maxpool_layer;
@ -15,8 +16,8 @@ typedef struct {
image get_maxpool_image(maxpool_layer layer);
maxpool_layer *make_maxpool_layer(int batch, int h, int w, int c, int size, int stride);
void resize_maxpool_layer(maxpool_layer *layer, int h, int w, int c);
void forward_maxpool_layer(const maxpool_layer layer, float *in);
void backward_maxpool_layer(const maxpool_layer layer, float *in, float *delta);
void forward_maxpool_layer(const maxpool_layer layer, float *input);
void backward_maxpool_layer(const maxpool_layer layer, float *input, float *delta);
#endif