probably how maxpool layers should be

This commit is contained in:
Joseph Redmon
2014-08-08 12:04:15 -07:00
parent b32a287e38
commit d9f1b0b16e
32 changed files with 1044 additions and 746 deletions

View File

@ -7,12 +7,13 @@ typedef struct {
int batch;
int h,w,c;
int stride;
int size;
float *delta;
float *output;
} maxpool_layer;
image get_maxpool_image(maxpool_layer layer);
maxpool_layer *make_maxpool_layer(int batch, int h, int w, int c, int stride);
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);