pull/5299/head
Joseph Redmon 9 years ago
parent ab75d5c578
commit 6bfd380bc1
  1. 4
      Makefile
  2. 10
      src/coco_demo.c
  3. 4
      src/yolo_kernels.cu

@ -1,9 +1,9 @@
GPU=0
OPENCV=1
CUDNN=0
OPENCV=0
DEBUG=0
ARCH= --gpu-architecture=compute_52 --gpu-code=compute_52
ARCH= --gpu-architecture=compute_52 --gpu-code=compute_52
VPATH=./src/
EXEC=darknet

@ -87,8 +87,6 @@ void demo_coco(char *cfgfile, char *weightfile, float thresh, int cam_index, con
}
if(!cap) error("Couldn't connect to webcam.\n");
cvNamedWindow("COCO", CV_WINDOW_NORMAL);
cvResizeWindow("COCO", 512, 512);
detection_layer l = net.layers[net.n-1];
int j;
@ -123,16 +121,22 @@ void demo_coco(char *cfgfile, char *weightfile, float thresh, int cam_index, con
}
int count = 0;
cvNamedWindow("COCO", CV_WINDOW_NORMAL);
cvMoveWindow("COCO", 0, 0);
cvResizeWindow("COCO", 1352, 1013);
while(1){
++count;
struct timeval tval_before, tval_after, tval_result;
gettimeofday(&tval_before, NULL);
if(pthread_create(&fetch_thread, 0, fetch_in_thread_coco, 0)) error("Thread creation failed");
if(pthread_create(&detect_thread, 0, detect_in_thread_coco, 0)) error("Thread creation failed");
//show_image(disp, "COCO");
show_image(disp, "COCO");
/*
char buff[256];
sprintf(buff, "/home/pjreddie/coco/coco_%05d", count);
save_image(disp, buff);
*/
free_image(disp);
cvWaitKey(10);

@ -99,6 +99,10 @@ extern "C" void demo_yolo(char *cfgfile, char *weightfile, float thresh, int cam
det = in;
det_s = in_s;
cvNamedWindow("YOLO", CV_WINDOW_NORMAL);
cvMoveWindow("YOLO", 0, 0);
cvResizeWindow("YOLO", 1352, 1013);
while(1){
struct timeval tval_before, tval_after, tval_result;
gettimeofday(&tval_before, NULL);

Loading…
Cancel
Save