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);