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

It takes into account the Difficult for calculating mAP for PascalVOC

Browse Source
This commit is contained in:
AlexeyAB
2018-02-16 23:55:37 +03:00
parent a1af57d8d6
commit 65bff2683b
9 changed files with 179 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/softmax_layer.c
View File

@ -40,7 +40,7 @@ void softmax_tree(float *input, int batch, int inputs, float temp, tree *hierarc
int count = 0;
for(i = 0; i < hierarchy->groups; ++i){
int group_size = hierarchy->group_size[i];
softmax(input+b*inputs + count, group_size, temp, output+b*inputs + count);
softmax(input+b*inputs + count, group_size, temp, output+b*inputs + count, 1);
count += group_size;
}
}
@ -55,7 +55,7 @@ void forward_softmax_layer(const softmax_layer l, network_state state)
softmax_tree(state.input, batch, inputs, l.temperature, l.softmax_tree, l.output);
} else {
for(b = 0; b < batch; ++b){
softmax(state.input+b*inputs, inputs, l.temperature, l.output+b*inputs);
softmax(state.input+b*inputs, inputs, l.temperature, l.output+b*inputs, 1);
}
}
}