diff --git a/src/parser.c b/src/parser.c index 184d1de3..1a324078 100644 --- a/src/parser.c +++ b/src/parser.c @@ -825,6 +825,11 @@ network parse_network_cfg_custom(char *filename, int batch) 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; }