mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
Fixed maxpool after change
This commit is contained in:
parent
96b578db63
commit
c39dfd188a
@ -27,14 +27,14 @@ maxpool_layer make_maxpool_layer(int batch, int h, int w, int c, int size, int s
|
|||||||
l.h = h;
|
l.h = h;
|
||||||
l.w = w;
|
l.w = w;
|
||||||
l.c = c;
|
l.c = c;
|
||||||
l.out_h = h;
|
l.out_h = (h-1)/stride + 1;
|
||||||
l.out_w = w;
|
l.out_w = (w-1)/stride + 1;
|
||||||
l.out_c = c;
|
l.out_c = c;
|
||||||
l.outputs = l.out_h * l.out_w * l.out_c;
|
l.outputs = l.out_h * l.out_w * l.out_c;
|
||||||
l.inputs = l.outputs;
|
l.inputs = l.outputs;
|
||||||
l.size = size;
|
l.size = size;
|
||||||
l.stride = stride;
|
l.stride = stride;
|
||||||
int output_size = ((h-1)/stride+1) * ((w-1)/stride+1) * c * batch;
|
int output_size = l.out_h * l.out_w * l.out_c * batch;
|
||||||
l.indexes = calloc(output_size, sizeof(int));
|
l.indexes = calloc(output_size, sizeof(int));
|
||||||
l.output = calloc(output_size, sizeof(float));
|
l.output = calloc(output_size, sizeof(float));
|
||||||
l.delta = calloc(output_size, sizeof(float));
|
l.delta = calloc(output_size, sizeof(float));
|
||||||
|
Loading…
Reference in New Issue
Block a user