YOLO源码学习(一)

    在学习完YOLOv1论文后,我比较想了解为什么通过这样的网络就能预测出来一张图片里物体的位置坐标,带着好奇的目的我开始了我YOLO源码的学习。

    大家都知道YOLO训练所要求的数据格式是PASCAL VOC或者COCO等标准数据集格式,而我们通常对图片标注后产生的xml文件,我们要将其转化为txt格式,而Darknet为我们提供了转换文件为voc_label.py。现在我们开始学习该程序,为我们下一步获取训练集的数据集做准备。在此我将一句一句进行理解,做解析。

1 获取VOC2012数据集

    首先获取VOC2012数据集,YOLO官网为我们提供了路径:

wget https://pjreddie.com/media/files/VOCtrainval_11-May-2012.tar


2 VOC2012数据集分析

    VOC数据集包括训练集、测试集,包含20个种类。

    数据集的组成架构分为:Annotations,ImageSets,JPEGImages,SegmentationClass,SegmentationObject五个文件夹,分别对5个文件夹进行解析。文件夹如图所示:


2.1 Annotations文件夹解析

    Annotations文件夹包含着所有图片信息的.xml信息,每个.xml文件代表着每一张图片。.xml文件的格式如图所示。后面讲述代码会分析.xml文件内容,读取.xml文件并将其转化为txt格式(YOLO可以使用的txt文档)

2.2 ImageSets

    ImageSets下有4个文件夹,我将对这4个文件夹的内容进行解析。4个文件夹如图所示:

Action文件夹中是包含着人的的动作(如jumping,walking,riding等)

Layout文件夹中是包含着具体人的部位的数据(如人的head,feet等)

Main文件夹中是图像是图像识别的数据,总共分为20类

Segmentation文件夹中是可用于分割的数据

2.3 JPEGImages

    JPEGImages就是存放着我们的所有数据集的图片.

2.4 SegmentationClass

    这里面包含了2913张图片,图片的像素有20种,即代表了20个种类。

2.5 SegmentationObject

    这里面包含了2913张图片,与SegmentationClass不同之处在于该是针对Object的,即一张图片中出现多个相同物体用不同的颜色来表示。

3 YOLO数据格式

    前面博文已经讲述Darknet有专门的py文件来将.xml文件转为.txt文件。即voc_label.py文件,现在我要将该py文件进行解析。


总结:通过voc_label.py文件我们将xml文件的数据转化为YOLO所要求的.txt文档,并且得到我们在训练时的obj.data中训练集,测试集的位置的txt,便于我们实验用,但有时候我们标注完产生的不是xml文件而是txt文件,我们怎么做,我遇到过这种问题,下篇博客我为大家讲述出现这种情况我们的代码解析,以及实际训练结果.

参考博客:https://blog.csdn.net/zhangjunbob/article/details/52769381

                








需要学习Windows系统YOLOv4的同学请前往《Windows版YOLOv4目标检测实战:原理与源码解析》,课程链接 https://edu.csdn.net/course/detail/29865【为什么要学习这门课】 Linux创始人Linus Torvalds有一句名言:Talk is cheap. Show me the code. 冗谈不够,放码过来!  代码阅读是从基础到提高的必由之路。尤其对深度学习,许多框架隐藏了神经网络底层的实现,只能在上层调包使用,对其内部原理很难认识清晰,不利于进一步优化和创新。YOLOv4是最近推出的基于深度学习的端到端实时目标检测方法。YOLOv4的实现darknet是使用C语言开发的轻型开源深度学习框架,依赖少,可移植性好,可以作为很好的代码阅读案例,让我们深入探究其实现原理。【课程内容与收获】 本课程将解析YOLOv4的实现原理和源码,具体内容包括:- YOLOv4目标检测原理- 神经网络及darknet的C语言实现,尤其是反向传播的梯度求解和误差计算- 代码阅读工具及方法- 深度学习计算的利器:BLAS和GEMM- GPU的CUDA编程方法及在darknet的应用- YOLOv4的程序流程- YOLOv4各层及关键技术的源码解析本课程将提供注释后的darknet的源码程序文件。【相关课程】 除本课程《YOLOv4目标检测:原理与源码解析》外,本人推出了有关YOLOv4目标检测的系列课程,包括:《YOLOv4目标检测实战:训练自己的数据集》《YOLOv4-tiny目标检测实战:训练自己的数据集》《YOLOv4目标检测实战:人脸口罩佩戴检测》《YOLOv4目标检测实战:中国交通标志识别》建议先学习一门YOLOv4实战课程,对YOLOv4的使用方法了解以后再学习本课程。【YOLOv4网络模型架构图】 下图由白勇老师绘制  
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值