From 920de66bfa85a9e694d78b6138114c723c854d7c Mon Sep 17 00:00:00 2001 From: AlexeyAB Date: Sat, 16 Nov 2019 17:23:21 +0300 Subject: [PATCH] gaussian_yolo: added uc_normalizer and minor fix for iou_normalizer for GIoU. --- src/parser.c | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/parser.c b/src/parser.c index f0533f36e12..3728c4422f9 100644 --- a/src/parser.c +++ b/src/parser.c @@ -444,8 +444,9 @@ layer parse_gaussian_yolo(list *options, size_params params) // Gaussian_YOLOv3 //assert(l.outputs == params.inputs); l.scale_x_y = option_find_float_quiet(options, "scale_x_y", 1); + l.uc_normalizer = option_find_float_quiet(options, "uc_normalizer", 1.0); l.iou_normalizer = option_find_float_quiet(options, "iou_normalizer", 0.75); - l.cls_normalizer = option_find_float_quiet(options, "cls_normalizer", 1); + l.cls_normalizer = option_find_float_quiet(options, "cls_normalizer", 1.0); char *iou_loss = option_find_str_quiet(options, "iou_loss", "mse"); // "iou"); if (strcmp(iou_loss, "mse") == 0) l.iou_loss = MSE;