blame | history | patch | commit | commitdiff | ignore whitespace
AlexeyAB
2018-04-19 c2c8595b083ec3586a99bb913b8a986e81e3a42a 
 src/detector.c


@@ -91,7 +91,7 @@


   args.small_object = net.small_object;


    args.d = &buffer;


    args.type = DETECTION_DATA;


   args.threads = 64;   // 8


   args.threads = 16;   // 64




    args.angle = net.angle;


    args.exposure = net.exposure;


@@ -99,6 +99,7 @@


    args.hue = net.hue;




#ifdef OPENCV


   args.threads = 3;


   IplImage* img = NULL;


   float max_img_loss = 5;


   int number_of_lines = 100;


@@ -108,7 +109,7 @@


#endif   //OPENCV




    pthread_t load_thread = load_data(args);


    clock_t time;


    double time;


    int count = 0;


    //while(i*imgs < N*120){


    while(get_current_batch(net) < net.max_batches){


@@ -131,7 +132,7 @@


            }


            net = nets[0];


        }


        time=clock();


        time=what_time_is_it_now();


        pthread_join(load_thread, 0);


        train = buffer;


        load_thread = load_data(args);


@@ -153,9 +154,9 @@


           save_image(im, "truth11");


         */




        printf("Loaded: %lf seconds\n", sec(clock()-time));


        printf("Loaded: %lf seconds\n", (what_time_is_it_now()-time));




        time=clock();


        time=what_time_is_it_now();


        float loss = 0;


#ifdef GPU


        if(ngpus == 1){


@@ -166,11 +167,11 @@


#else


        loss = train_network(net, train);


#endif


        if (avg_loss < 0) avg_loss = loss;


        if (avg_loss < 0 || avg_loss != avg_loss) avg_loss = loss;   // if(-inf or nan)


        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), sec(clock()-time), i*imgs);


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




#ifdef OPENCV


      if(!dont_show)


@@ -291,11 +292,11 @@


   int *map = 0;


   if (mapf) map = read_map(mapf);




   network net = parse_network_cfg_custom(cfgfile, 1);


   network net = parse_network_cfg_custom(cfgfile, 1);   // set batch=1


   if (weightfile) {


      load_weights(&net, weightfile);


   }


   set_batch_network(&net, 1);


   //set_batch_network(&net, 1);


   fprintf(stderr, "Learning Rate: %g, Momentum: %g, Decay: %g\n", net.learning_rate, net.momentum, net.decay);


   srand(time(0));




@@ -385,7 +386,7 @@


         int nboxes = 0;


         int letterbox = (args.type == LETTERBOX_DATA);


         detection *dets = get_network_boxes(&net, w, h, thresh, .5, map, 0, &nboxes, letterbox);


         if (nms) do_nms_sort_v3(dets, nboxes, classes, nms);


         if (nms) do_nms_sort(dets, nboxes, classes, nms);


         if (coco) {


            print_cocos(fp, path, dets, nboxes, classes, w, h);


         }


@@ -414,11 +415,12 @@




void validate_detector_recall(char *datacfg, char *cfgfile, char *weightfile)


{


   network net = parse_network_cfg_custom(cfgfile, 1);


   network net = parse_network_cfg_custom(cfgfile, 1);   // set batch=1


   if (weightfile) {


      load_weights(&net, weightfile);


   }


   set_batch_network(&net, 1);


   //set_batch_network(&net, 1);


   fuse_conv_batchnorm(net);


   srand(time(0));




   //list *plist = get_paths("data/coco_val_5k.list");


@@ -452,12 +454,15 @@


      int nboxes = 0;


      int letterbox = 0;


      detection *dets = get_network_boxes(&net, sized.w, sized.h, thresh, .5, 0, 1, &nboxes, letterbox);


      if (nms) do_nms_obj_v3(dets, nboxes, 1, nms);


      if (nms) do_nms_obj(dets, nboxes, 1, nms);




      char labelpath[4096];


      find_replace(path, "images", "labels", labelpath);


      find_replace(labelpath, "JPEGImages", "labels", labelpath);


      find_replace(labelpath, ".jpg", ".txt", labelpath);


      find_replace(labelpath, ".png", ".txt", labelpath);


      find_replace(labelpath, ".bmp", ".txt", labelpath);


      find_replace(labelpath, ".JPG", ".txt", labelpath);


      find_replace(labelpath, ".JPEG", ".txt", labelpath);




      int num_labels = 0;


@@ -482,7 +487,7 @@


            ++correct;


         }


      }




      //fprintf(stderr, " %s - %s - ", paths[i], labelpath);


      fprintf(stderr, "%5d %5d %5d\tRPs/Img: %.2f\tIOU: %.2f%%\tRecall:%.2f%%\n", i, correct, total, (float)proposals / (i + 1), avg_iou * 100 / total, 100.*correct / total);


      free(id);


      free_image(orig);


@@ -521,11 +526,12 @@


   int *map = 0;


   if (mapf) map = read_map(mapf);




   network net = parse_network_cfg_custom(cfgfile, 1);


   network net = parse_network_cfg_custom(cfgfile, 1);   // set batch=1


   if (weightfile) {


      load_weights(&net, weightfile);


   }


   set_batch_network(&net, 1);


   //set_batch_network(&net, 1);


   fuse_conv_batchnorm(net);


   srand(time(0));




   list *plist = get_paths(valid_images);


@@ -603,15 +609,18 @@


         int nboxes = 0;


         int letterbox = (args.type == LETTERBOX_DATA);


         float hier_thresh = 0;


         detection *dets = get_network_boxes(&net, 1, 1, thresh, hier_thresh, 0, 1, &nboxes, letterbox);


         if (nms) do_nms_sort_v3(dets, nboxes, l.classes, nms);


         detection *dets = get_network_boxes(&net, 1, 1, thresh, hier_thresh, 0, 0, &nboxes, letterbox);


         //detection *dets = get_network_boxes(&net, val[t].w, val[t].h, thresh, hier_thresh, 0, 1, &nboxes, letterbox); // for letterbox=1


         if (nms) do_nms_sort(dets, nboxes, l.classes, nms);




         char labelpath[4096];


         find_replace(path, "images", "labels", labelpath);


         find_replace(labelpath, "JPEGImages", "labels", labelpath);


         find_replace(labelpath, ".jpg", ".txt", labelpath);


         find_replace(labelpath, ".JPEG", ".txt", labelpath);


         find_replace(labelpath, ".png", ".txt", labelpath);


         find_replace(labelpath, ".bmp", ".txt", labelpath);


         find_replace(labelpath, ".JPG", ".txt", labelpath);


         find_replace(labelpath, ".JPEG", ".txt", labelpath);


         int num_labels = 0;


         box_label *truth = read_boxes(labelpath, &num_labels);


         int i, j;


@@ -857,8 +866,10 @@


      find_replace(path, "images", "labels", labelpath);


      find_replace(labelpath, "JPEGImages", "labels", labelpath);


      find_replace(labelpath, ".jpg", ".txt", labelpath);


      find_replace(labelpath, ".JPEG", ".txt", labelpath);


      find_replace(labelpath, ".png", ".txt", labelpath);


      find_replace(labelpath, ".bmp", ".txt", labelpath);


      find_replace(labelpath, ".JPG", ".txt", labelpath);


      find_replace(labelpath, ".JPEG", ".txt", labelpath);


      int num_labels = 0;


      box_label *truth = read_boxes(labelpath, &num_labels);


      //printf(" new path: %s \n", labelpath);


@@ -1017,13 +1028,14 @@


    char **names = get_labels(name_list);




    image **alphabet = load_alphabet();


    network net = parse_network_cfg_custom(cfgfile, 1);


    network net = parse_network_cfg_custom(cfgfile, 1); // set batch=1


    if(weightfile){


        load_weights(&net, weightfile);


    }


    set_batch_network(&net, 1);


    //set_batch_network(&net, 1);


   fuse_conv_batchnorm(net);


    srand(2222222);


    clock_t time;


    double time;


    char buff[256];


    char *input = buff;


    int j;


@@ -1050,15 +1062,16 @@


        //for(j = 0; j < l.w*l.h*l.n; ++j) probs[j] = calloc(l.classes, sizeof(float *));




        float *X = sized.data;


        time=clock();


        network_predict(net, X);


        printf("%s: Predicted in %f seconds.\n", input, sec(clock()-time));


        time= what_time_is_it_now();


        //network_predict(net, X);


      network_predict_image(&net, im); letterbox = 1;


        printf("%s: Predicted in %f seconds.\n", input, (what_time_is_it_now()-time));


        //get_region_boxes(l, 1, 1, thresh, probs, boxes, 0, 0);


      // if (nms) do_nms_sort(boxes, probs, l.w*l.h*l.n, l.classes, nms);


      // if (nms) do_nms_sort_v2(boxes, probs, l.w*l.h*l.n, l.classes, nms);


      //draw_detections(im, l.w*l.h*l.n, thresh, boxes, probs, names, alphabet, l.classes);


      int nboxes = 0;


      detection *dets = get_network_boxes(&net, im.w, im.h, thresh, hier_thresh, 0, 1, &nboxes, letterbox);


      if (nms) do_nms_sort_v3(dets, nboxes, l.classes, nms);


      if (nms) do_nms_sort(dets, nboxes, l.classes, nms);


      draw_detections_v3(im, dets, nboxes, thresh, names, alphabet, l.classes);


      free_detections(dets, nboxes);


        save_image(im, "predictions");


@@ -1146,4 +1159,5 @@


        demo(cfg, weights, thresh, hier_thresh, cam_index, filename, names, classes, frame_skip, prefix, out_filename,


         http_stream_port, dont_show);


    }


   else printf(" There isn't such command: %s", argv[2]);


}