let's break some shit

This commit is contained in:
Joseph Redmon 2017-05-09 15:57:22 -07:00
parent 40c9afa381
commit 25a3e687a3
2 changed files with 5 additions and 4 deletions

View File

@ -1,6 +1,6 @@
GPU=0 GPU=1
CUDNN=0 CUDNN=1
OPENCV=0 OPENCV=1
DEBUG=0 DEBUG=0
ARCH= -gencode arch=compute_20,code=[sm_20,sm_21] \ 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] -gencode arch=compute_52,code=[sm_52,compute_52]
# This is what I use, uncomment if you know your arch and want to specify # 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/ VPATH=./src/
EXEC=darknet EXEC=darknet

View File

@ -274,6 +274,7 @@ layer parse_region(list *options, size_params params)
l.sqrt = option_find_int_quiet(options, "sqrt", 0); l.sqrt = option_find_int_quiet(options, "sqrt", 0);
l.softmax = option_find_int(options, "softmax", 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.max_boxes = option_find_int_quiet(options, "max",30);
l.jitter = option_find_float(options, "jitter", .2); l.jitter = option_find_float(options, "jitter", .2);
l.rescore = option_find_int_quiet(options, "rescore",0); l.rescore = option_find_int_quiet(options, "rescore",0);