blame | history | patch | commit | commitdiff | ignore whitespace
Alexey
2018-05-31 0713fd9784bbd0c67eb82df6f6e2c072a53e09e8 
 src/yolo.c


@@ -191,7 +191,7 @@


            int w = val[t].w;


            int h = val[t].h;


            get_detection_boxes(l, w, h, thresh, probs, boxes, 0);


            if (nms) do_nms_sort(boxes, probs, l.side*l.side*l.n, classes, iou_thresh);


            if (nms) do_nms_sort_v2(boxes, probs, l.side*l.side*l.n, classes, iou_thresh);


            print_yolo_detections(fps, id, boxes, probs, l.side*l.side*l.n, classes, w, h);


            free(id);


            free_image(val[t]);


@@ -252,10 +252,7 @@


        if (nms) do_nms(boxes, probs, side*side*l.n, 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, ".JPEG", ".txt", labelpath);


      replace_image_to_label(path, labelpath);




        int num_labels = 0;


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


@@ -322,7 +319,7 @@


        network_predict(net, X);


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


        get_detection_boxes(l, 1, 1, thresh, probs, boxes, 0);


        if (nms) do_nms_sort(boxes, probs, l.side*l.side*l.n, l.classes, nms);


        if (nms) do_nms_sort_v2(boxes, probs, l.side*l.side*l.n, l.classes, nms);


        //draw_detections(im, l.side*l.side*l.n, thresh, boxes, probs, voc_names, alphabet, 20);


        draw_detections(im, l.side*l.side*l.n, thresh, boxes, probs, voc_names, alphabet, 20);


        save_image(im, "predictions");


@@ -348,6 +345,7 @@


   float hier_thresh = find_float_arg(argc, argv, "-hier", .5);


    int cam_index = find_int_arg(argc, argv, "-c", 0);


    int frame_skip = find_int_arg(argc, argv, "-s", 0);


   int ext_output = find_arg(argc, argv, "-ext_output");


    if(argc < 4){


        fprintf(stderr, "usage: %s %s [train/test/valid] [cfg] [weights (optional)]\n", argv[0], argv[1]);


        return;


@@ -361,5 +359,5 @@


    else if(0==strcmp(argv[2], "valid")) validate_yolo(cfg, weights);


    else if(0==strcmp(argv[2], "recall")) validate_yolo_recall(cfg, weights);


    else if(0==strcmp(argv[2], "demo")) demo(cfg, weights, thresh, hier_thresh, cam_index, filename, voc_names, 20, frame_skip,


      prefix, out_filename, http_stream_port, dont_show);


      prefix, out_filename, http_stream_port, dont_show, ext_output);


}