mirror of https://github.com/AlexeyAB/darknet.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
21 lines
538 B
21 lines
538 B
#ifndef MAXPOOL_LAYER_H |
|
#define MAXPOOL_LAYER_H |
|
|
|
#include "image.h" |
|
|
|
typedef struct { |
|
int batch; |
|
int h,w,c; |
|
int stride; |
|
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); |
|
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); |
|
|
|
#endif |
|
|
|
|