Merge pull request #2542 from kossolax/patch-2

Segfault if height > width
This commit is contained in:
Alexey
2019-03-13 02:41:42 +03:00
committed by GitHub

View File

@ -100,7 +100,7 @@ void train_classifier(char *datacfg, char *cfgfile, char *weightfile, int *gpus,
args.exposure = net.exposure;
args.saturation = net.saturation;
args.hue = net.hue;
args.size = net.w;
args.size = net.w > net.h ? net.w : net.h;
args.paths = paths;
args.classes = classes;