From 25a3e687a3627ac10682aa1a577adff03c0e0f2e Mon Sep 17 00:00:00 2001 From: Joseph Redmon Date: Tue, 9 May 2017 15:57:22 -0700 Subject: [PATCH] let's break some shit --- Makefile | 8 ++++---- src/parser.c | 1 + 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/Makefile b/Makefile index 029d02cb..4f12f70b 100644 --- a/Makefile +++ b/Makefile @@ -1,6 +1,6 @@ -GPU=0 -CUDNN=0 -OPENCV=0 +GPU=1 +CUDNN=1 +OPENCV=1 DEBUG=0 ARCH= -gencode arch=compute_20,code=[sm_20,sm_21] \ @@ -10,7 +10,7 @@ ARCH= -gencode arch=compute_20,code=[sm_20,sm_21] \ -gencode arch=compute_52,code=[sm_52,compute_52] # This is what I use, uncomment if you know your arch and want to specify -# ARCH= -gencode arch=compute_52,code=compute_52 +ARCH= -gencode arch=compute_52,code=compute_52 VPATH=./src/ EXEC=darknet diff --git a/src/parser.c b/src/parser.c index 4add3573..a1cf5f7d 100644 --- a/src/parser.c +++ b/src/parser.c @@ -274,6 +274,7 @@ layer parse_region(list *options, size_params params) l.sqrt = option_find_int_quiet(options, "sqrt", 0); l.softmax = option_find_int(options, "softmax", 0); + l.background = option_find_int_quiet(options, "background", 0); l.max_boxes = option_find_int_quiet(options, "max",30); l.jitter = option_find_float(options, "jitter", .2); l.rescore = option_find_int_quiet(options, "rescore",0);