diff --git a/src/region_layer.c b/src/region_layer.c index 7aa1a196..b7aba32e 100644 --- a/src/region_layer.c +++ b/src/region_layer.c @@ -20,6 +20,10 @@ region_layer make_region_layer(int batch, int w, int h, int n, int classes, int l.batch = batch; l.h = h; l.w = w; + l.c = n*(classes + coords + 1); + l.out_w = l.w; + l.out_h = l.h; + l.out_c = l.c; l.classes = classes; l.coords = coords; l.cost = (float*)xcalloc(1, sizeof(float));