mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
Merge pull request #2542 from kossolax/patch-2
Segfault if height > width
This commit is contained in:
@ -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;
|
||||
|
Reference in New Issue
Block a user