Joseph Redmon
2015-11-17 b394337824c59dd970b87c5978fe4e8b94b85cb2
src/yolo_kernels.cu
@@ -61,7 +61,7 @@
    return 0;
}
extern "C" void demo_yolo(char *cfgfile, char *weightfile, float thresh)
extern "C" void demo_yolo(char *cfgfile, char *weightfile, float thresh, int cam_index)
{
    printf("YOLO demo\n");
    net = parse_network_cfg(cfgfile);
@@ -72,7 +72,7 @@
    srand(2222222);
    cv::VideoCapture cam(0);
    cv::VideoCapture cam(cam_index);
    cap = cam;
    if(!cap.isOpened()) error("Couldn't connect to webcam.\n");
@@ -118,7 +118,7 @@
    }
}
#else
extern "C" void demo_yolo(char *cfgfile, char *weightfile, float thresh){
extern "C" void demo_yolo(char *cfgfile, char *weightfile, float thresh, int cam_index){
    fprintf(stderr, "YOLO demo needs OpenCV for webcam images.\n");
}
#endif