From fed6d6e31d819ecae21e9c3b9315b5877deb3f42 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 22 Sep 2015 01:18:51 +0000
Subject: [PATCH] idk man
---
src/parser.c | 12 ++++++++----
1 files changed, 8 insertions(+), 4 deletions(-)
diff --git a/src/parser.c b/src/parser.c
index 53e8461..7ea1b3f 100644
--- a/src/parser.c
+++ b/src/parser.c
@@ -182,10 +182,14 @@
int num = option_find_int(options, "num", 1);
int side = option_find_int(options, "side", 7);
region_layer layer = make_region_layer(params.batch, params.inputs, num, side, classes, coords, rescore);
- int softmax = option_find_int(options, "softmax", 0);
- int sqrt = option_find_int(options, "sqrt", 0);
- layer.softmax = softmax;
- layer.sqrt = sqrt;
+
+ layer.softmax = option_find_int(options, "softmax", 0);
+ layer.sqrt = option_find_int(options, "sqrt", 0);
+
+ layer.coord_scale = option_find_float(options, "coord_scale", 1);
+ layer.object_scale = option_find_float(options, "object_scale", 1);
+ layer.noobject_scale = option_find_float(options, "noobject_scale", 1);
+ layer.class_scale = option_find_float(options, "class_scale", 1);
return layer;
}
--
Gitblit v1.10.0