Joseph Redmon
2015-02-24 5f4a5f59b072d4029107422d30b04941424c48b1
src/cost_layer.c
@@ -49,7 +49,7 @@
    if(layer.type == DETECTION){
        int i;
        for(i = 0; i < layer.batch*layer.inputs; ++i){
            if((i%5) && !truth[(i/5)*5]) layer.delta[i] = 0;
            if((i%25) && !truth[(i/25)*25]) layer.delta[i] = 0;
        }
    }
    *(layer.output) = dot_cpu(layer.batch*layer.inputs, layer.delta, 1, layer.delta, 1);
@@ -71,7 +71,7 @@
    axpy_ongpu(layer.batch*layer.inputs, -1, input, 1, layer.delta_gpu, 1);
    if(layer.type==DETECTION){
        mask_ongpu(layer.inputs*layer.batch, layer.delta_gpu, truth, 5);
        mask_ongpu(layer.inputs*layer.batch, layer.delta_gpu, truth, 25);
    }
    cuda_pull_array(layer.delta_gpu, layer.delta, layer.batch*layer.inputs);