OrgLion-ML
/
darknet
mirror of https://github.com/AlexeyAB/darknet.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Add YOLO Layer 'nbiases' (#8893) 

Co-authored-by: KyeongSoo Kim <kyeongsoo.kim@konantech.com>
This commit is contained in:
Kyeongsoo Kim(KnsoF) 2024-05-03 22:53:53 +09:00 committed by GitHub
parent 08167076fd
commit d02cc3a816
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/yolo_layer.c
View File

@ -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.classes = classes;
l.cost = (float*)xcalloc(1, sizeof(float)); l.cost = (float*)xcalloc(1, sizeof(float));
l.biases = (float*)xcalloc(total * 2, sizeof(float)); l.biases = (float*)xcalloc(total * 2, sizeof(float));
l.nbiases = total * 2;
if(mask) l.mask = mask; if(mask) l.mask = mask;
else{ else{
l.mask = (int*)xcalloc(n, sizeof(int)); l.mask = (int*)xcalloc(n, sizeof(int));