mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
gonna change im2col
This commit is contained in:
@ -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;
|
||||
}
|
||||
|
Reference in New Issue
Block a user