yolov3调用mysql_yolo源码解析（3）：视频检测流程

最新推荐文章于 2023-05-24 21:13:09 发布

珂珂爱啊

最新推荐文章于 2023-05-24 21:13:09 发布

阅读量198

点赞数

文章标签： yolov3调用mysql

本文链接：https://blog.csdn.net/weixin_35411439/article/details/113495168

版权

本文详细介绍了如何使用Yolov3进行视频检测，从调用命令到源码解析，探讨了CPU处理视频的局限性和解决方法。同时，展示了如何在Ubuntu系统上安装libmysqlclient-dev库，并提供了一个简单的MySQL连接C++示例，成功实现与数据库的连接。

摘要由CSDN通过智能技术生成

代码在自己电脑中！！！！不在服务器

根据前文所说yolo代码逻辑：

├── examples

│ ├── darknet.c(主程序)

│ │── xxx1.c

│ └── xxx2.c

│

├── include

│ ├── darknet.h

│

├── Makefile

│

└── src

├── yyy1.c

├── yyy2.h

└──......

视频检测入口时darknet.c文件。

输入视频检测命令，如： ./darknet detector demo cfg/coco.data cfg/yolov3.cfg yolov3.weights ~/dataset/ca3.mp4 ,darknet.c中根据detector参数，调用detector.c中的 run_detector(int argc, char **argv); 函数.

run_detector函数内容如下

void run_detector(int argc, char **argv)

{char *prefix = find_char_arg(argc, argv, "-prefix", 0);float thresh = find_float_arg(argc, argv, "-thresh", .5);float hier_thresh = find_float_arg(argc, argv, "-hier", .5);int cam_index = find_int_arg(argc, argv, "-c", 0);int frame_skip = find_int_arg(argc, argv, "-s", 0);int avg = find_int_arg(argc, argv, "-avg", 3);if(argc < 4){

fprintf(stderr,"usage: %s %s [train/test/valid] [cfg] [weights (optional)]\n", argv[0], argv[1]);return;

}char *gpu_list = find_char_arg(argc, argv, "-gpus", 0);char *outfile = find_char_arg(argc, argv, "-out", 0);int *gpus = 0;int gpu = 0;int ngpus = 0;if(gpu_list){

printf("%s\n", gpu_list);int len =strlen(gpu_list);

ngpus= 1;inti;for(i = 0; i < len; ++i){if (gpu_list[i] == ',') ++ngpus;

}

gpus= calloc(ngpus, sizeof(int));for(i = 0; i < ngpus; ++i){

gpus[i]=atoi(gpu_list);

gpu_list= strchr(gpu_list, ',')+1;

}

}else{

gpu=gpu_index;

gpus= &gpu;

ngpus= 1;

}int clear = find_arg(argc, argv, "-clear");int fullscreen = find_arg(argc, argv, "-fullscreen");int width = find_int_arg(argc, argv, "-w", 0);int height = find_int_arg(argc, argv, "-h", 0);int fps = find_int_arg(argc, argv, "-fps", 0);//int class = find_int_arg(argc, argv, "-class", 0);

View Code

代码中定义了frame_skip等变量，默认值为0.然后继续调用demo函数，demo函数位于/darknet/src/demo.c文件中，内容如下。

void demo(char *cfgfile, char *weightfile, float thresh, int cam_index, const char *filename, char **names, int classes, int delay, char *prefix, int avg_frames, float hier, int w, int h, int frames, intfullscreen)

{//demo_frame = avg_frames;

image **alphabet =load_alphabet();

demo_names=names;

demo_alphabet=alphabet;

demo_classes=classes;

demo_thresh=thresh;

demo_hier=hier;

printf("Demo\n");

net= load_network(cfgfile, weightfile, 0);

set_batch_network(net,1);

pthread_t detect_thread;

pthread_t fetch_thread;

srand(2222222);inti;

demo_total=size_network(net);

predictions= calloc(demo_frame, sizeof(float*));for (i = 0; i < demo_frame; ++i){

predictions[i]= calloc(demo_total, sizeof(float));

}

avg= calloc(demo_total, sizeof(float));if(filename){

printf("video file: %s\n", filename);

cap= open_video_stream(filename, 0, 0, 0, 0);

}else{

cap= open_video_stream(0, cam_index, w, h, frames);

}if(!cap) error("Couldn't connect to webcam.\n");

buff[0] =get_image_from_stream(cap);

buff[1] = copy_image(buff[0]);

buff[2] = copy_image(buff[0]);

buff_letter[0] = letterbox_image(buff[0], net->w, net->h);

buff_letter[1] = letterbox_image(buff[0], net->w, net->h);

buff_letter[2] = letterbox_image(buff[0], net->w, net->h);int count = 0;if(!prefix){

make_window("Demo", 1352, 1013, fullscreen);

}

demo_time=what_time_is_it_now();while(!demo_done){

buff_index= (buff_index + 1) %3;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(!prefix){

fps= 1./(what_time_is_it_now() -demo_time);

demo_time=what_time_is_it_now();

display_in_thread(0);

}else{char name[256];

sprintf(name,"%s_%08d", prefix, count);

save_image(buff[(buff_index+ 1)%3], name);

}

pthread_join(fetch_thread,0);

pthread_join(detect_thread,0);++count;

}

View Code

yolo视频处理的参考：https://blog.csdn.net/sinat_33718563/article/details/79964758