From c53e03348c65462bcba33f6352087dd6b9268e8f Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Wed, 16 Sep 2015 21:12:10 +0000
Subject: [PATCH] yolo working w/ regions
---
src/parser.c | 4 ++++
1 files changed, 4 insertions(+), 0 deletions(-)
diff --git a/src/parser.c b/src/parser.c
index 94dc0fa..53e8461 100644
--- a/src/parser.c
+++ b/src/parser.c
@@ -182,6 +182,10 @@
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;
return layer;
}
--
Gitblit v1.10.0