up new v4

Scheaven

2021-06-03 168af40fe9a3cc81c6ee16b3e81f154780c36bdb

 lib/detecter_tools/darknet/demo.c

@@ -36,13 +36,12 @@
static int demo_ext_output = 0;

static long long int frame_id = 0;

static int demo_json_port = -1;

static bool demo_skip_frame = false;


#define NFRAMES 3



static float* predictions[NFRAMES];

static int avg_frames;
static int demo_index = 0;

static mat_cv* cv_images[NFRAMES];

static float *avg;

static mat_cv** cv_images;


mat_cv* in_img;

mat_cv* det_img;

@@ -61,6 +60,8 @@
    while (!custom_atomic_load_int(&flag_exit)) {

        while (!custom_atomic_load_int(&run_fetch_in_thread)) {

            if (custom_atomic_load_int(&flag_exit)) return 0;

            if (demo_skip_frame)
                consume_frame(cap);
            this_thread_yield();

        }

        int dont_close_stream = 0;    // set 1 if your IP-camera periodically turns off and turns on video-stream

@@ -99,22 +100,23 @@


        layer l = net.layers[net.n - 1];

        float *X = det_s.data;

        float *prediction = network_predict(net, X);



        int i;

        for (i = 0; i < net.n; ++i) {

            layer l = net.layers[i];

            if (l.type == YOLO) l.mean_alpha = 1.0 / NFRAMES;

        }

        //float *prediction =
        network_predict(net, X);


        cv_images[demo_index] = det_img;

        det_img = cv_images[(demo_index + NFRAMES / 2 + 1) % NFRAMES];

        demo_index = (demo_index + 1) % NFRAMES;

        det_img = cv_images[(demo_index + avg_frames / 2 + 1) % avg_frames];
        demo_index = (demo_index + 1) % avg_frames;


        if (letter_box)

            dets = get_network_boxes(&net, get_width_mat(in_img), get_height_mat(in_img), demo_thresh, demo_thresh, 0, 1, &nboxes, 1); // letter box

        else

            dets = get_network_boxes(&net, net.w, net.h, demo_thresh, demo_thresh, 0, 1, &nboxes, 0); // resized


        //const float nms = .45;
        //if (nms) {
        //    if (l.nms_kind == DEFAULT_NMS) do_nms_sort(dets, nboxes, l.classes, nms);
        //    else diounms_sort(dets, nboxes, l.classes, nms, l.nms_kind, l.beta_nms);
        //}


        custom_atomic_store_int(&run_detect_in_thread, 0);

    }

@@ -138,10 +140,12 @@
    return (double)walltime.tv_sec + (double)walltime.tv_usec * .000001;

}



void demo(char *cfgfile, char *weightfile, float thresh, float hier_thresh, int cam_index, const char *filename, char **names, int classes,

void demo(char *cfgfile, char *weightfile, float thresh, float hier_thresh, int cam_index, const char *filename, char **names, int classes, int avgframes,
    int frame_skip, char *prefix, char *out_filename, int mjpeg_port, int dontdraw_bbox, int json_port, int dont_show, int ext_output, int letter_box_in, int time_limit_sec, char *http_post_host,

    int benchmark, int benchmark_layers)

{

    if (avgframes < 1) avgframes = 1;
    avg_frames = avgframes;
    letter_box = letter_box_in;

    in_img = det_img = show_img = NULL;

    //skip = frame_skip;

@@ -158,6 +162,7 @@
    if(weightfile){

        load_weights(&net, weightfile);

    }

    if (net.letter_box) letter_box = 1;
    net.benchmark_layers = benchmark_layers;

    fuse_conv_batchnorm(net);

    calculate_binary_weights(net);

@@ -166,9 +171,11 @@
    if(filename){

        printf("video file: %s\n", filename);

        cap = get_capture_video_stream(filename);

        demo_skip_frame = is_live_stream(filename);
    }else{

        printf("Webcam index: %d\n", cam_index);

        cap = get_capture_webcam(cam_index);

        demo_skip_frame = true;
    }



    if (!cap) {

@@ -181,8 +188,16 @@
    layer l = net.layers[net.n-1];

    int j;



    avg = (float *) calloc(l.outputs, sizeof(float));

    for(j = 0; j < NFRAMES; ++j) predictions[j] = (float *) calloc(l.outputs, sizeof(float));

    cv_images = (mat_cv**)xcalloc(avg_frames, sizeof(mat_cv));

    int i;
    for (i = 0; i < net.n; ++i) {
        layer lc = net.layers[i];
        if (lc.type == YOLO) {
            lc.mean_alpha = 1.0 / avg_frames;
            l = lc;
        }
    }


    if (l.classes != demo_classes) {

        printf("\n Parameters don't match: in cfg-file classes=%d, in data-file classes=%d \n", l.classes, demo_classes);

@@ -206,7 +221,7 @@
    det_img = in_img;

    det_s = in_s;



    for (j = 0; j < NFRAMES / 2; ++j) {

    for (j = 0; j < avg_frames / 2; ++j) {
        free_detections(dets, nboxes);

        fetch_in_thread_sync(0); //fetch_in_thread(0);

        detect_in_thread_sync(0); //fetch_in_thread(0);

@@ -263,6 +278,8 @@
                else diounms_sort(local_dets, local_nboxes, l.classes, nms, l.nms_kind, l.beta_nms);

            }



            if (l.embedding_size) set_track_id(local_dets, local_nboxes, demo_thresh, l.sim_thresh, l.track_ciou_norm, l.track_history_size, l.dets_for_track, l.dets_for_show);

            //printf("\033[2J");

            //printf("\033[1;1H");

            //printf("\nFPS:%.1f\n", fps);

@@ -292,7 +309,7 @@


            if(!prefix){

                if (!dont_show) {

                    const int each_frame = max_val_cmp(1, avg_fps / 100);

                    const int each_frame = max_val_cmp(1, avg_fps / 60);
                    if(global_frame_counter % each_frame == 0) show_image_mat(show_img, "Demo");

                    int c = wait_key_cv(1);

                    if (c == 10) {

@@ -389,16 +406,14 @@
    free_image(in_s);

    free_detections(dets, nboxes);



    free(avg);

    for (j = 0; j < NFRAMES; ++j) free(predictions[j]);

    demo_index = (NFRAMES + demo_index - 1) % NFRAMES;

    for (j = 0; j < NFRAMES; ++j) {

    demo_index = (avg_frames + demo_index - 1) % avg_frames;
    for (j = 0; j < avg_frames; ++j) {
            release_mat(&cv_images[j]);

    }

    free(cv_images);


    free_ptrs((void **)names, net.layers[net.n - 1].classes);



    int i;

    const int nsize = 8;

    for (j = 0; j < nsize; ++j) {

        for (i = 32; i < 127; ++i) {

@@ -411,7 +426,7 @@
    //cudaProfilerStop();

}

#else

void demo(char *cfgfile, char *weightfile, float thresh, float hier_thresh, int cam_index, const char *filename, char **names, int classes,

void demo(char *cfgfile, char *weightfile, float thresh, float hier_thresh, int cam_index, const char *filename, char **names, int classes, int avgframes,
    int frame_skip, char *prefix, char *out_filename, int mjpeg_port, int dontdraw_bbox, int json_port, int dont_show, int ext_output, int letter_box_in, int time_limit_sec, char *http_post_host,

    int benchmark, int benchmark_layers)

{