From d02cc3a81689a1593dc5d0c8ed018d10338bebdd Mon Sep 17 00:00:00 2001 From: "Kyeongsoo Kim(KnsoF)" <56542593+K9714@users.noreply.github.com> Date: Fri, 3 May 2024 22:53:53 +0900 Subject: [PATCH] Add YOLO Layer 'nbiases' (#8893) Co-authored-by: KyeongSoo Kim --- src/yolo_layer.c | 1 + 1 file changed, 1 insertion(+) diff --git a/src/yolo_layer.c b/src/yolo_layer.c index ac464ba39b4..de9d0990455 100644 --- a/src/yolo_layer.c +++ b/src/yolo_layer.c @@ -29,6 +29,7 @@ layer make_yolo_layer(int batch, int w, int h, int n, int total, int *mask, int l.classes = classes; l.cost = (float*)xcalloc(1, sizeof(float)); l.biases = (float*)xcalloc(total * 2, sizeof(float)); + l.nbiases = total * 2; if(mask) l.mask = mask; else{ l.mask = (int*)xcalloc(n, sizeof(int));