Added Focal Loss to yolo-layer

This commit is contained in:
AlexeyAB
2018-04-05 23:27:02 +03:00
parent be9d971ddb
commit 943f6e874b
5 changed files with 51 additions and 11 deletions

View File

@ -274,6 +274,7 @@ layer parse_yolo(list *options, size_params params)
//l.max_boxes = option_find_int_quiet(options, "max", 90);
l.jitter = option_find_float(options, "jitter", .2);
l.focal_loss = option_find_int_quiet(options, "focal_loss", 0);
l.ignore_thresh = option_find_float(options, "ignore_thresh", .5);
l.truth_thresh = option_find_float(options, "truth_thresh", 1);