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.exposure = net.exposure;
|
||||||
args.saturation = net.saturation;
|
args.saturation = net.saturation;
|
||||||
args.hue = net.hue;
|
args.hue = net.hue;
|
||||||
args.size = net.w;
|
args.size = net.w > net.h ? net.w : net.h;
|
||||||
|
|
||||||
args.paths = paths;
|
args.paths = paths;
|
||||||
args.classes = classes;
|
args.classes = classes;
|
||||||
|
Reference in New Issue
Block a user