gonna change im2col

This commit is contained in:
Joseph Redmon
2015-03-21 12:25:14 -07:00
parent dcb000b553
commit 4af116e996
16 changed files with 153 additions and 78 deletions

View File

@ -165,7 +165,8 @@ detection_layer *parse_detection(list *options, size_params params)
int coords = option_find_int(options, "coords", 1);
int classes = option_find_int(options, "classes", 1);
int rescore = option_find_int(options, "rescore", 1);
detection_layer *layer = make_detection_layer(params.batch, params.inputs, classes, coords, rescore);
int background = option_find_int(options, "background", 1);
detection_layer *layer = make_detection_layer(params.batch, params.inputs, classes, coords, rescore, background);
option_unused(options);
return layer;
}