Fixed some problems with visualizing networks

Joseph Redmon

2015-07-24 fdcc096694791f92d2cff0e4fd74fbabfca12936

 src/network.c

@@ -125,13 +125,20 @@

float get_network_cost(network net)
{
    int i;
    float sum = 0;
    int count = 0;
    for(i = 0; i < net.n; ++i){
    if(net.layers[net.n-1].type == COST){
        return net.layers[net.n-1].output[0];
            sum += net.layers[net.n-1].output[0];
            ++count;
    }
    if(net.layers[net.n-1].type == DETECTION){
        return net.layers[net.n-1].cost[0];
            sum += net.layers[net.n-1].cost[0];
            ++count;
    }
    return 0;
    }
    return sum/count;
}

int get_predicted_class_network(network net)