mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
First Commit!
This commit is contained in:
24
src/maxpool_layer.c
Normal file
24
src/maxpool_layer.c
Normal file
@ -0,0 +1,24 @@
|
||||
#include "maxpool_layer.h"
|
||||
|
||||
maxpool_layer make_maxpool_layer(int h, int w, int c, int stride)
|
||||
{
|
||||
maxpool_layer layer;
|
||||
layer.stride = stride;
|
||||
layer.output = make_image((h-1)/stride+1, (w-1)/stride+1, c);
|
||||
return layer;
|
||||
}
|
||||
|
||||
void run_maxpool_layer(const image input, const maxpool_layer layer)
|
||||
{
|
||||
int i,j,k;
|
||||
for(i = 0; i < layer.output.h*layer.output.w*layer.output.c; ++i) layer.output.data[i] = -DBL_MAX;
|
||||
for(i = 0; i < input.h; ++i){
|
||||
for(j = 0; j < input.w; ++j){
|
||||
for(k = 0; k < input.c; ++k){
|
||||
double val = get_pixel(input, i, j, k);
|
||||
double cur = get_pixel(layer.output, i/layer.stride, j/layer.stride, k);
|
||||
if(val > cur) set_pixel(layer.output, i/layer.stride, j/layer.stride, k, val);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user