mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
Fixes for training Yolo on small objects
This commit is contained in:
@ -82,6 +82,7 @@ void train_detector(char *datacfg, char *cfgfile, char *weightfile, int *gpus, i
|
||||
args.classes = classes;
|
||||
args.jitter = jitter;
|
||||
args.num_boxes = l.max_boxes;
|
||||
args.small_object = l.small_object;
|
||||
args.d = &buffer;
|
||||
args.type = DETECTION_DATA;
|
||||
args.threads = 4;// 8;
|
||||
|
Reference in New Issue
Block a user