From c6ecf1e0420737eafeb99b27b1d716b46a6cbb7a Mon Sep 17 00:00:00 2001
From: Jud White <github@judsonwhite.com>
Date: Sun, 25 Mar 2018 20:41:48 +0000
Subject: [PATCH] README.md: add notes to How to compile on Windows

---
 src/demo.c |    6 ++++--
 1 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/demo.c b/src/demo.c
index 59672f8..b5917fa 100644
--- a/src/demo.c
+++ b/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();
 

--
Gitblit v1.10.0