真正的darknet源码解析之一

今天开始,记录darknet源码的解读。 

darknet 官网 https://pjreddie.com/darknet/yolo/

进入darknet目录会出现下面的目录结构 

backup:主要存放训练的模型 ;cfg:网络模型的结构;data:一些数据其中label文件夹作用是在检测结构最后显示label的结果如Person等;example:存放了一些.c文件其中比较主要的是darknet.c和detector.c;

include:包含darknet.h 头文件,主要是一些定义和 函数声明;src:绝大部分的源码,各种层等等。

Start!

下面就官网的测试命令作为入口

./darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg
./darknet 是执行当前文件下面已经编译好的darknet文件
detect 是命令 后面三个分别是参数: 网络模型 网络权重 需要检测的图片
那么这个命令在哪呢?

argv[1] 在上面的命令中就是 detect 可以看出这里调用了test_detector()

对比上面可以看出 *datacfg被默认为cfg/coco.data  .data存放的是你训练集(train)/测试集(valid) 图片的绝对路径 以及类别数目(classes)以及类别名称文件夹(names = data/coco.names) 

这个函数是检测的重点部分 (敲黑板!)

 go on~

 具体的network_predict()过程我就不再赘述了 gdb调试的话也就是数据的不断前馈,最后给出网络的结果

我们主要来看get_network_boxes()

这次就讲到这里,下次说的是网络前馈之后出来的数据到底是什么样的形式?怎么解析的?各种值x,y,w,h,c,classes都是怎么存放的



猜你喜欢

转载自www.cnblogs.com/codebirdhan/p/9440100.html
今日推荐