src/detection_layer.c
@@ -58,8 +58,8 @@ int index = b*l.inputs; for (i = 0; i < locations; ++i) { int offset = i*l.classes; softmax_array(l.output + index + offset, l.classes, 1, l.output + index + offset); softmax(l.output + index + offset, l.classes, 1, l.output + index + offset, 1); } } }