mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
AlexeyAB
2018-06-15 17:25:58 +03:00
parent 6ed2474e14
commit e066cad1ec
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/parser.c
View File

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