Jud White
2018-03-25 c6ecf1e0420737eafeb99b27b1d716b46a6cbb7a
src/detector.c
@@ -43,9 +43,6 @@
    float avg_loss = -1;
    network *nets = calloc(ngpus, sizeof(network));
   int iter_save;
   iter_save = 100;
    srand(time(0));
    int seed = rand();
    int i;
@@ -79,6 +76,8 @@
   int init_w = net.w;
   int init_h = net.h;
   int iter_save;
   iter_save = get_current_batch(net);
    load_args args = {0};
    args.w = net.w;
@@ -92,7 +91,7 @@
   args.small_object = l.small_object;
    args.d = &buffer;
    args.type = DETECTION_DATA;
   args.threads = 8; // 64
   args.threads = 64;   // 8
    args.angle = net.angle;
    args.exposure = net.exposure;
@@ -180,8 +179,8 @@
      //if (i % 1000 == 0 || (i < 1000 && i % 100 == 0)) {
      //if (i % 100 == 0) {
      if(i >= iter_save) {
         iter_save += 100;
      if(i >= (iter_save + 100)) {
         iter_save = i;
#ifdef GPU
         if (ngpus != 1) sync_nets(nets, ngpus, 0);
#endif
@@ -1032,6 +1031,7 @@
        }
        image im = load_image_color(input,0,0);
        image sized = resize_image(im, net.w, net.h);
      //image sized = letterbox_image(im, net.w, net.h);
        layer l = net.layers[net.n-1];
        box *boxes = calloc(l.w*l.h*l.n, sizeof(box));