AlexeyAB
2018-06-04 efaf684cb20b996db0cd6d99d20fabb4bc26859a
src/detector.c
@@ -87,7 +87,8 @@
    load_args args = {0};
    args.w = net.w;
    args.h = net.h;
    args.paths = paths;
   args.c = net.c;
   args.paths = paths;
    args.n = imgs;
    args.m = plist->size;
    args.classes = classes;
@@ -105,7 +106,7 @@
    args.hue = net.hue;
#ifdef OPENCV
   args.threads = 3;
   args.threads = 3 * ngpus;
   IplImage* img = NULL;
   float max_img_loss = 5;
   int number_of_lines = 100;
@@ -189,7 +190,7 @@
        avg_loss = avg_loss*.9 + loss*.1;
        i = get_current_batch(net);
        printf("\n %d: %f, %f avg, %f rate, %lf seconds, %d images\n", get_current_batch(net), loss, avg_loss, get_current_rate(net), (what_time_is_it_now()-time), i*imgs);
        printf("\n %d: %f, %f avg loss, %f rate, %lf seconds, %d images\n", get_current_batch(net), loss, avg_loss, get_current_rate(net), (what_time_is_it_now()-time), i*imgs);
#ifdef OPENCV
      if(!dont_show)
@@ -388,6 +389,7 @@
   load_args args = { 0 };
   args.w = net.w;
   args.h = net.h;
   args.c = net.c;
   args.type = IMAGE_DATA;
   //args.type = LETTERBOX_DATA;
@@ -482,7 +484,7 @@
   for (i = 0; i < m; ++i) {
      char *path = paths[i];
      image orig = load_image_color(path, 0, 0);
      image orig = load_image(path, 0, 0, net.c);
      image sized = resize_image(orig, net.w, net.h);
      char *id = basecfg(path);
      network_predict(net, sized.data);
@@ -595,6 +597,7 @@
   load_args args = { 0 };
   args.w = net.w;
   args.h = net.h;
   args.c = net.c;
   args.type = IMAGE_DATA;
   //args.type = LETTERBOX_DATA;
@@ -1093,10 +1096,10 @@
            if(!input) return;
            strtok(input, "\n");
        }
        image im = load_image_color(input,0,0);
        image im = load_image(input,0,0,net.c);
      int letterbox = 0;
        //image sized = resize_image(im, net.w, net.h);
      image sized = letterbox_image(im, net.w, net.h); letterbox = 1;
        image sized = resize_image(im, net.w, net.h);
      //image sized = letterbox_image(im, net.w, net.h); letterbox = 1;
        layer l = net.layers[net.n-1];
        //box *boxes = calloc(l.w*l.h*l.n, sizeof(box));