mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

gaussian_yolo: added uc_normalizer and minor fix for iou_normalizer for GIoU.

Browse Source
This commit is contained in:
AlexeyAB
2019-11-16 17:23:21 +03:00
parent 0d30db35fc
commit 920de66bfa
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/parser.c
View File

@ -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;