./darknet detector valid uses letter_box=1 from cfg-file [net] letter_box=1

pull/5550/head
AlexeyAB 5 years ago
parent da050367a8
commit b5870c12d6
  1. 6
      src/detector.c

@ -696,7 +696,8 @@ void validate_detector(char *datacfg, char *cfgfile, char *weightfile, char *out
args.h = net.h;
args.c = net.c;
args.type = IMAGE_DATA;
//args.type = LETTERBOX_DATA;
const int letter_box = net.letter_box;
if (letter_box) args.type = LETTERBOX_DATA;
for (t = 0; t < nthreads; ++t) {
args.path = paths[i + t];
@ -726,8 +727,7 @@ void validate_detector(char *datacfg, char *cfgfile, char *weightfile, char *out
int w = val[t].w;
int h = val[t].h;
int nboxes = 0;
int letterbox = (args.type == LETTERBOX_DATA);
detection *dets = get_network_boxes(&net, w, h, thresh, .5, map, 0, &nboxes, letterbox);
detection *dets = get_network_boxes(&net, w, h, thresh, .5, map, 0, &nboxes, letter_box);
if (nms) {
if (l.nms_kind == DEFAULT_NMS) do_nms_sort(dets, nboxes, l.classes, nms);
else diounms_sort(dets, nboxes, l.classes, nms, l.nms_kind, l.beta_nms);

Loading…
Cancel
Save