mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

yolo demo low latency

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
cfg/xyolo.test.cfg
View File

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

11
src/yolo_demo.c
View File

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