From 8f1b4e0962857d402f9d017fcbf387ef0eceb7c4 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Thu, 01 Sep 2016 23:48:41 +0000
Subject: [PATCH] updates and things
---
src/demo.c | 24 +++++++++++++++---------
1 files changed, 15 insertions(+), 9 deletions(-)
diff --git a/src/demo.c b/src/demo.c
index 2e7ab93..020513f 100644
--- a/src/demo.c
+++ b/src/demo.c
@@ -8,7 +8,7 @@
#include "demo.h"
#include <sys/time.h>
-#define FRAMES 1
+#define FRAMES 3
#ifdef OPENCV
#include "opencv2/highgui/highgui_c.h"
@@ -48,7 +48,7 @@
void *detect_in_thread(void *ptr)
{
- float nms = .4;
+ float nms = .1;
detection_layer l = net.layers[net.n-1];
float *X = det_s.data;
@@ -153,13 +153,19 @@
if(pthread_create(&fetch_thread, 0, fetch_in_thread, 0)) error("Thread creation failed");
if(pthread_create(&detect_thread, 0, detect_in_thread, 0)) error("Thread creation failed");
- show_image(disp, "Demo");
- int c = cvWaitKey(1);
- if (c == 10){
- if(frame_skip == 0) frame_skip = 60;
- else if(frame_skip == 4) frame_skip = 0;
- else if(frame_skip == 60) frame_skip = 4;
- else frame_skip = 0;
+ if(1){
+ show_image(disp, "Demo");
+ int c = cvWaitKey(1);
+ if (c == 10){
+ if(frame_skip == 0) frame_skip = 60;
+ else if(frame_skip == 4) frame_skip = 0;
+ else if(frame_skip == 60) frame_skip = 4;
+ else frame_skip = 0;
+ }
+ }else{
+ char buff[256];
+ sprintf(buff, "/home/pjreddie/tmp/bag_%07d", count);
+ save_image(disp, buff);
}
pthread_join(fetch_thread, 0);
--
Gitblit v1.10.0