gaussian_yolo: added uc_normalizer and minor fix for iou_normalizer f…

…or GIoU.
AlexeyAB · Nov 16, 2019 · 920de66 · 920de66
1 parent 0d30db3
commit 920de66
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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;