Joseph Redmon
2015-04-15 d97331b88ff3d50035b1e22c9d0eb671b61227e3
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)