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

This commit is contained in:
Ievgen Popovych 2021-06-11 03:36:12 +03:00 committed by GitHub
parent 7ff80b1a47
commit 894aada70b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 0 deletions

View File

@ -20,6 +20,10 @@ region_layer make_region_layer(int batch, int w, int h, int n, int classes, int
l.batch = batch; l.batch = batch;
l.h = h; l.h = h;
l.w = w; 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.classes = classes;
l.coords = coords; l.coords = coords;
l.cost = (float*)xcalloc(1, sizeof(float)); l.cost = (float*)xcalloc(1, sizeof(float));