| | |
| | | static image det_s; |
| | | static image disp = {0}; |
| | | static CvCapture * cap; |
| | | static int use_webcam = 0; |
| | | static float fps = 0; |
| | | static float demo_thresh = 0; |
| | | |
| | |
| | | |
| | | void draw_detections_cv(IplImage* show_img, int num, float thresh, box *boxes, float **probs, char **names, image **alphabet, int classes); |
| | | void show_image_cv_ipl(IplImage *disp, const char *name); |
| | | image get_image_from_stream_resize(CvCapture *cap, int w, int h, IplImage** in_img); |
| | | image get_image_from_stream_resize(CvCapture *cap, int w, int h, IplImage** in_img, int use_webcam); |
| | | IplImage* in_img; |
| | | IplImage* det_img; |
| | | IplImage* show_img; |
| | |
| | | void *fetch_in_thread(void *ptr) |
| | | { |
| | | //in = get_image_from_stream(cap); |
| | | in = get_image_from_stream_resize(cap, net.w, net.h, &in_img); |
| | | in = get_image_from_stream_resize(cap, net.w, net.h, &in_img, use_webcam); |
| | | if(!in.data){ |
| | | //error("Stream closed."); |
| | | flag_exit = 1; |
| | |
| | | printf("video file: %s\n", filename); |
| | | cap = cvCaptureFromFile(filename); |
| | | }else{ |
| | | printf("Webcam index: %d\n", cam_index); |
| | | #ifdef CV_VERSION_EPOCH // OpenCV 2.x |
| | | cap = cvCaptureFromCAM(cam_index); |
| | | #else // OpenCV 3.x |
| | | use_webcam = 1; |
| | | cap = get_capture_webcam(cam_index); |
| | | #endif |
| | | } |
| | | |
| | | if(!cap) error("Couldn't connect to webcam.\n"); |
| | |
| | | } |
| | | |
| | | // if you run it with param -http_port 8090 then open URL in your web-browser: http://localhost:8090 |
| | | if (http_stream_port > 0) { |
| | | if (http_stream_port > 0 && show_img) { |
| | | //int port = 8090; |
| | | int port = http_stream_port; |
| | | int timeout = 200; |
| | |
| | | } |
| | | |
| | | // save video file |
| | | if (output_video_writer) { |
| | | if (output_video_writer && show_img) { |
| | | cvWriteFrame(output_video_writer, show_img); |
| | | printf("\n cvWriteFrame \n"); |
| | | } |
| | | |
| | | cvReleaseImage(&show_img); |
| | | |
| | | pthread_join(fetch_thread, 0); |