AlexeyAB
2018-06-15 e066cad1ec730cea0fa76348e34c15ff637c6975
Added warning if width or height isn't divisible by 32 for default Yolo-networks
1 files modified
5 ■■■■■ changed files
src/parser.c 5 ●●●●● patch | view | raw | blame | history
src/parser.c
@@ -825,6 +825,11 @@
        net.workspace = calloc(1, workspace_size);
#endif
    }
    LAYER_TYPE lt = net.layers[net.n - 1].type;
    if ((net.w % 32 != 0 || net.h % 32 != 0) && (lt == YOLO || lt == REGION || lt == DETECTION)) {
        printf("\n Warning: width=%d and height=%d in cfg-file must be divisible by 32 for default networks Yolo v1/v2/v3!!! \n\n",
            net.w, net.h);
    }
    return net;
}