From 5d0aa6ec522afd5d859c068f40eded77d6704fa0 Mon Sep 17 00:00:00 2001 From: AlexeyAB Date: Sun, 17 Nov 2019 01:18:28 +0300 Subject: [PATCH] Compile fix --- src/gaussian_yolo_layer.c | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/gaussian_yolo_layer.c b/src/gaussian_yolo_layer.c index 16971437..93569cf0 100644 --- a/src/gaussian_yolo_layer.c +++ b/src/gaussian_yolo_layer.c @@ -374,7 +374,7 @@ void forward_gaussian_yolo_layer(const layer l, network_state state) int class_index = entry_gaussian_index(l, b, n*l.w*l.h + j*l.w + i, 9); int obj_index = entry_gaussian_index(l, b, n*l.w*l.h + j*l.w + i, 8); float objectness = l.output[obj_index]; - int class_id_match = compare_yolo_class(l.output, l.classes, class_index, l.w*l.h, objectness, class_id, 0.25f); + int class_id_match = compare_gaussian_yolo_class(l.output, l.classes, class_index, l.w*l.h, objectness, class_id, 0.25f); float iou = box_iou(pred, truth); if (iou > best_match_iou && class_id_match == 1) {