mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

Classifier now is working. The file softmax_layer.c was outdated

Browse Source
This commit is contained in:
erick
2018-10-06 14:12:02 -05:00
parent 24b60456c7
commit 38801979b2
8 changed files with 219 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/parser.c
View File

@ -233,12 +233,17 @@ connected_layer parse_connected(list *options, size_params params)
softmax_layer parse_softmax(list *options, size_params params)
{
int groups = option_find_int_quiet(options, "groups",1);
softmax_layer layer = make_softmax_layer(params.batch, params.inputs, groups);
layer.temperature = option_find_float_quiet(options, "temperature", 1);
char *tree_file = option_find_str(options, "tree", 0);
if (tree_file) layer.softmax_tree = read_tree(tree_file);
return layer;
int groups = option_find_int_quiet(options, "groups", 1);
softmax_layer layer = make_softmax_layer(params.batch, params.inputs, groups);
layer.temperature = option_find_float_quiet(options, "temperature", 1);
char *tree_file = option_find_str(options, "tree", 0);
if (tree_file) layer.softmax_tree = read_tree(tree_file);
layer.w = params.w;
layer.h = params.h;
layer.c = params.c;
layer.spatial = option_find_float_quiet(options, "spatial", 0);
layer.noloss = option_find_int_quiet(options, "noloss", 0);
return layer;
}
int *parse_yolo_mask(char *a, int *num)