yolov2 损失函数_yolo v2 损失函数源码解读

region_layer.c

box get_region_box(float *x, float *biases, int n, int index, int i, int j, int w, int h, int stride)

{

box b;

b.x = (i + x[index + 0*stride]) / w;

b.y = (j + x[index + 1*stride]) / h;

b.w = exp(x[index + 2*stride]) * biases[2*n]   / w;

b.h = exp(x[index + 3*stride]) * biases[2*n+1] / h;

//printf("%f/%d/%d - %f/%f/%f/%f\n", x[index + 2*stride], w, h, b.x, b.y, b.w, b.h);

return b;

}

float delta_region_box(box truth, float *x, float *biases, int n, int index, int i, int j, int w, int h, float *delta, float scale, int stride)

{

box pred = get_region_box(x, biases, n, index, i, j, w, h, stride);

float iou = box_iou(pred, truth);

float tx = (truth.x*w - i);

float ty = (truth.y*h - j);

float tw = log(truth.w*w / biases[2*n]);

float th = log(truth.h*h / biases[2*n + 1]);

delta[index + 0*stride] = scale * (tx - x[index + 0*stride]);

delta[index + 1*stride] = scale * (ty - x[index + 1*stride]);

delta[index + 2*stride] = scale * (tw - x[index + 2*stride]);

delta[index + 3*stride] = scale * (th - x[index + 3*stride]);

return iou;

}

void forward_region_layer()

{

...

for (b = 0; b 

if(l.softmax_tree){

// 没执行

}

// 下面的 for 循环是计算没有物体的 box 的 confidence 的 loss

// 1， 遍历所有格子以及每个格子的 box，计算每个 box 与真实 box 的 best_iou

// 2， 先不管三七二十一，把该 box