Fixes for training Yolo on small objects

This commit is contained in:
AlexeyAB
2018-02-08 01:04:40 +03:00
parent 64aa0180bb
commit 9ff40baa4f
5 changed files with 16 additions and 5 deletions

View File

@ -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;