Jud White
2018-03-25 c6ecf1e0420737eafeb99b27b1d716b46a6cbb7a
src/demo.c
@@ -64,6 +64,7 @@
   in = get_image_from_stream_resize(cap, net.w, net.h, &in_img, use_webcam);
    if(!in.data){
        //error("Stream closed.");
      printf("Stream closed.\n");
      flag_exit = 1;
      return;
    }
@@ -166,6 +167,8 @@
    probs = (float **)calloc(l.w*l.h*l.n, sizeof(float *));
    for(j = 0; j < l.w*l.h*l.n; ++j) probs[j] = (float *)calloc(l.classes, sizeof(float *));
   flag_exit = 0;
    pthread_t fetch_thread;
    pthread_t detect_thread;
@@ -198,7 +201,7 @@
    }
   CvVideoWriter* output_video_writer = NULL;    // cv::VideoWriter output_video;
   if (out_filename)
   if (out_filename && !flag_exit)
   {
      CvSize size;
      size.width = det_img->width, size.height = det_img->height;
@@ -212,7 +215,6 @@
      //output_video_writer = cvCreateVideoWriter(out_filename, CV_FOURCC('X', 'V', 'I', 'D'), 25, size, 1);
      //output_video_writer = cvCreateVideoWriter(out_filename, CV_FOURCC('W', 'M', 'V', '2'), 25, size, 1);
   }
   flag_exit = 0;
    double before = get_wall_time();