yolo demo low latency

This commit is contained in:
Joseph Redmon 2016-06-19 22:37:21 -07:00
parent 08c7cf9c88
commit a45d17ffe3
2 changed files with 16 additions and 10 deletions

View File

@ -13,15 +13,6 @@ steps=20,40,60,80,20000,30000
scales=5,5,2,2,.1,.1
max_batches = 40000
[crop]
crop_width=448
crop_height=448
flip=0
angle=0
saturation = 1.5
exposure = 1.5
noadjust=1
[convolutional]
batch_normalize=1
filters=16
@ -112,6 +103,7 @@ stride=2
[batchnorm]
[convolutional]
xnor = 1
batch_normalize=1
filters=1024
size=3
@ -119,12 +111,15 @@ stride=1
pad=1
activation=leaky
[batchnorm]
[convolutional]
xnor = 1
batch_normalize=1
filters=256
size=3
stride=1
pad=1
filters=128
activation=leaky
[connected]

View File

@ -98,6 +98,7 @@ void demo_yolo(char *cfgfile, char *weightfile, float thresh, int cam_index, cha
while(1){
struct timeval tval_before, tval_after, tval_result;
gettimeofday(&tval_before, NULL);
/*
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, "YOLO");
@ -109,6 +110,16 @@ void demo_yolo(char *cfgfile, char *weightfile, float thresh, int cam_index, cha
disp = det;
det = in;
det_s = in_s;
*/
fetch_in_thread(0);
det = in;
det_s = in_s;
detect_in_thread(0);
disp = det;
show_image(disp, "YOLO");
free_image(disp);
cvWaitKey(1);
gettimeofday(&tval_after, NULL);
timersub(&tval_after, &tval_before, &tval_result);