It takes into account the Difficult for calculating mAP for PascalVOC

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

View File

@ -59,7 +59,7 @@ void forward_detection_layer(const detection_layer l, network_state state)
for (i = 0; i < locations; ++i) {
int offset = i*l.classes;
softmax(l.output + index + offset, l.classes, 1,
l.output + index + offset);
l.output + index + offset, 1);
}
}
}