darknet之region_layer源码分析

void forward_region_layer(const region_layer l, network_state state)
{
   
    int i,j,b,t,n;
    // 25
    int size = l.coords + l.classes + 1;
    memcpy(l.output, state.input, l.outputs*l.batch*sizeof(float));
    #ifndef GPU
    flatten(l.output, l.w*l.h, size*l.n, l.batch, 1);
    #endif
    // 每个预测框的置信度
    for (b = 0; b < l.batch; ++b){
   
        for(i = 0; i < l.h*l.w*l.n; ++i){
   
            int index = size*i + b*l.outputs;
            l.output[index + 4] = logistic_activate(l.output[index + 4]);
        }
    }


#ifndef GPU
    if (l.softmax_tree){
   
        for (b = 0; b < l.batch; ++b){
   
            for(i = 0; i < l.h*l.w*l.n; ++i){
   
                int index = size*i + b*l.outputs;
                softmax_tree(l.output + index + 5, 1, 0, 1, l.softmax_tree, l.output + index + 5);
            }
        }
    } else if (l.softmax){
   
        for (b = 0; b < l.batch; ++b){
   
            for(i = 0; i < l.h*l.w*l.n; ++i){
   
                int index = size*i + b*l.outputs;
                softmax(l.output + index + 5, l.classes, 1, l.output + index + 5, 1);
            }
        }
    }
#endif
    if(!state.train) return;
    memset(l.delta, 0, l.outputs * l.batch * sizeof(float));
    float avg_iou = 0;
    float recall = 0;
    float avg_cat = 0;
    float avg_obj = 0;
    float avg_anyobj = 0