【darknet源码学习(一)】检测分类器训练过程源码解析

最新推荐文章于 2022-07-29 16:41:01 发布

页禾_Amie

最新推荐文章于 2022-07-29 16:41:01 发布

阅读量2.7k

点赞数 1

文章标签： yolov3 darknet

本文链接：https://blog.csdn.net/sinat_35862942/article/details/82424756

版权

本文从darknet的main()函数开始，详细解析了darknet检测分类器训练过程的源码。通过命令行参数启动训练，进入run_detector函数，当参数为'train'时，调用train_detector函数进行网络初始化、加载预训练权重、并行加载训练数据及训练网络的操作。

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

1.找main()函数

首先找到程序入口，在darknet.c中第351行处就是main()函数。利用生成的可执行文件进行训练时，输入的命令行参数为：

Linux下：./darknet detector train cfg/coco.data cfg/yolov3.cfg （darknet53.conv.74）（-gpus 0,1,2,3) 括号表示可选

windows下：与Linux类似，只是./darknet 变成darknet.exe

在main()中可以看到是通过argv[1]来选择下一步操作的，针对训练命令argv[1]是“detector”，看以下代码：

else if (0 == strcmp(argv[1], "detector")){
        run_detector(argc, argv);

程序跳转到run_detector(argc,argv)函数。

2.detector.c /run_detector()函数 1207行

void run_detector(int argc, char **argv)
{
    int dont_show = find_arg(argc, argv, "-dont_show");
    int show = find_arg(argc, argv, "-show");
    int http_stream_port = find_int_arg(argc, argv, "-http_port", -1);
    char *out_filename = find_char_arg(argc, argv, "-out_filename", 0);
    char *outfile = find_char_arg(argc, argv, "-out", 0);
    char *prefix = find_char_arg(argc, argv, "-prefix", 0);
    float thresh = find_float_arg(argc, argv, "-thresh", .25);    // 0.24
    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 num_of_clusters = find_int_arg(argc, argv, "-num_of_clusters", 5);
    int width = find_int_arg(argc, argv, "-width", -1);
    int height = find_int_arg(argc, argv, "-height", -1);
    // extended output in test mode (output of rect bound coords)
    // and for recall mode (extended output table-like format with results for best_class fit)
    int ext_output = find_arg(argc, argv, "-ext_output");
    int save_labels = find_arg(argc, argv, "-save_labels");
    if(argc < 4){
        fprintf(stderr, "usage: %s %s [train/test/valid] [cfg] [weights (optional)]\n", argv[0], argv[1]);
        return;
    }
    
    // 解析输入参数，获取GPU使用情况，如果使用单个GPU，那么调用时不需要指明GPU卡号，默认使用卡号0上的GPU;
    // 如果使用多块GPU，那么在调用时，其中有两个参数必须为：-gpus 0,1,2...（以逗号隔开）
    char *gpu_list = find_char_arg(argc, argv, "-gpus", 0);  //将用到的gpu列表0，1，2，3读 入gpu_list
    int *gpus = 0;  
    int gpu = 0;
    int ngpus = 0;  //使用的gpu个数
    if(gpu_list){
        printf("%s\n", gpu_list);
        int len = strlen(gpu_list);
        ngpus = 1;
        int