Joseph Redmon
2015-05-06 e7688a05a194e3c8baf3c11fbf09b7f5e8640a77
src/network.c
@@ -186,6 +186,9 @@
    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;
}