From 4b60afcc640a340a87fa56431322a4bb4ae1cfea Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 08 Nov 2016 07:42:19 +0000
Subject: [PATCH] stuff and things
---
src/parser.c | 4 +++-
1 files changed, 3 insertions(+), 1 deletions(-)
diff --git a/src/parser.c b/src/parser.c
index 26f45d3..4e71fe5 100644
--- a/src/parser.c
+++ b/src/parser.c
@@ -268,6 +268,7 @@
l.rescore = option_find_int_quiet(options, "rescore",0);
l.thresh = option_find_float(options, "thresh", .5);
+ l.classfix = option_find_int_quiet(options, "classfix", 0);
l.coord_scale = option_find_float(options, "coord_scale", 1);
l.object_scale = option_find_float(options, "object_scale", 1);
@@ -357,6 +358,7 @@
layer parse_reorg(list *options, size_params params)
{
int stride = option_find_int(options, "stride",1);
+ int reverse = option_find_int_quiet(options, "reverse",0);
int batch,h,w,c;
h = params.h;
@@ -365,7 +367,7 @@
batch=params.batch;
if(!(h && w && c)) error("Layer before reorg layer must output image.");
- layer layer = make_reorg_layer(batch,w,h,c,stride);
+ layer layer = make_reorg_layer(batch,w,h,c,stride,reverse);
return layer;
}
--
Gitblit v1.10.0