Joseph Redmon
2016-09-12 5c067dc44785a761a0243d8cd634e3ac17d548ad
src/yolo.c
@@ -54,6 +54,11 @@
    args.d = &buffer;
    args.type = REGION_DATA;
    args.angle = net.angle;
    args.exposure = net.exposure;
    args.saturation = net.saturation;
    args.hue = net.hue;
    pthread_t load_thread = load_data_in_thread(args);
    clock_t time;
    //while(i*imgs < N*120){
@@ -173,7 +178,7 @@
    int nms = 1;
    float iou_thresh = .5;
    int nthreads = 2;
    int nthreads = 8;
    image *val = calloc(nthreads, sizeof(image));
    image *val_resized = calloc(nthreads, sizeof(image));
    image *buf = calloc(nthreads, sizeof(image));
@@ -323,7 +328,7 @@
    char buff[256];
    char *input = buff;
    int j;
    float nms=.5;
    float nms=.4;
    box *boxes = calloc(l.side*l.side*l.n, sizeof(box));
    float **probs = calloc(l.side*l.side*l.n, sizeof(float *));
    for(j = 0; j < l.side*l.side*l.n; ++j) probs[j] = calloc(l.classes, sizeof(float *));
@@ -350,7 +355,6 @@
        save_image(im, "predictions");
        show_image(im, "predictions");
        show_image(sized, "resized");
        free_image(im);
        free_image(sized);
#ifdef OPENCV