From e066cad1ec730cea0fa76348e34c15ff637c6975 Mon Sep 17 00:00:00 2001 From: AlexeyAB Date: Fri, 15 Jun 2018 17:25:58 +0300 Subject: [PATCH] Added warning if width or height isn't divisible by 32 for default Yolo-networks --- src/parser.c | 5 +++++ 1 file changed, 5 insertions(+) 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; }