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

pull/1106/head
AlexeyAB 7 years ago
parent 6ed2474e14
commit e066cad1ec
  1. 5
      src/parser.c

@ -825,6 +825,11 @@ network parse_network_cfg_custom(char *filename, int batch)
net.workspace = calloc(1, workspace_size); net.workspace = calloc(1, workspace_size);
#endif #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; return net;
} }

Loading…
Cancel
Save