Added warning if width or height isn't divisible by 32 for default Yolo-networks

This commit is contained in:
AlexeyAB
2018-06-15 17:25:58 +03:00
parent 6ed2474e14
commit e066cad1ec

View File

@ -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;
}