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

IOU loss function

Browse Source
This commit is contained in:
Joseph Redmon
2015-04-24 10:27:50 -07:00
parent feabcc31de
commit 989ab8c38a
8 changed files with 272 additions and 62 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/network.c
View File

@ -186,6 +186,9 @@ float get_network_cost(network net)
if(net.types[net.n-1] == COST){
return ((cost_layer *)net.layers[net.n-1])->output[0];
}
if(net.types[net.n-1] == DETECTION){
return ((detection_layer *)net.layers[net.n-1])->cost[0];
}
return 0;
}