Fixes for training Yolo on small objects

This commit is contained in:
AlexeyAB
2018-02-08 01:04:40 +03:00
parent 64aa0180bb
commit 9ff40baa4f
5 changed files with 16 additions and 5 deletions

View File

@ -245,6 +245,7 @@ layer parse_region(list *options, size_params params)
l.log = option_find_int_quiet(options, "log", 0);
l.sqrt = option_find_int_quiet(options, "sqrt", 0);
l.small_object = option_find_int(options, "small_object", 0);
l.softmax = option_find_int(options, "softmax", 0);
l.max_boxes = option_find_int_quiet(options, "max",30);
l.jitter = option_find_float(options, "jitter", .2);