mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
Added warning if width or height isn't divisible by 32 for default Yolo-networks
This commit is contained in:
@ -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;
|
||||
}
|
||||
|
||||
|
Reference in New Issue
Block a user