IOU loss function

Joseph Redmon

2015-04-24 989ab8c38a02fa7ea9c25108151736c62e81c972

 src/normalization_layer.c

@@ -6,7 +6,7 @@
    int h = layer.h;
    int w = layer.w;
    int c = layer.c;
    return float_to_image(h,w,c,layer.output);
    return float_to_image(w,h,c,layer.output);
}

image get_normalization_delta(normalization_layer layer)
@@ -14,7 +14,7 @@
    int h = layer.h;
    int w = layer.w;
    int c = layer.c;
    return float_to_image(h,w,c,layer.delta);
    return float_to_image(w,h,c,layer.delta);
}

normalization_layer *make_normalization_layer(int batch, int h, int w, int c, int size, float alpha, float beta, float kappa)