region_layer: Populate out_h/w/c fileds for consitency (AlexeyAB#7783)

uf-mil · Jun 11, 2021 · 894aada · 894aada
1 parent 7ff80b1
commit 894aada
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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));