目标检测入门论文YOLOV1精读以及pytorch源码复现(yolov1)

2023-01-31 22:07:56

结果展示

其中绿线是我绘制的图像划分网格。

这里的loss是我训练的 0.77 ，由于损失函数是我自己写的，所以可能跟大家的不太一样，这个不重要，重要的是学习思路。

重点提示

yolov1是一个目标检测的算法，他是一阶段的检测算法。

一阶段（one-stage）：检测物体的同时进行分类。（代表论文：yolov1 - yolov5）

二阶段（two-stage）：先检测出物体，再进行分类。（代表论文：rcnn，fast-rcnn）

重点要理解yolov1的数据特征标注方式。

只有理解了数据特征的标注方式才可以理解他为什么可以起作用。

论文剖析

1、理解VOC数据集的数据形式。

2、从VOC数据集中提取出标注好的数据特征。

3、yolov1的数据组织。

4、yolov1的算法模型。

5、yolov1的准确率评估方式（IOU）。

6、yolov1的损失函数。

理解VOC数据集

首先需要知道我们使用的数据集的形式，因为每一个数据集的特征标注以及组织方式都不同。

我们可以去官网下载voc的数据集，这里使用的是voc2012数据集。

VOC数据集镜像网站. 下载voc2012的Train/Validation Data (1.9 GB)。

数据集下载之后解压出来是这样子：

每个文件夹存放的啥都标注好了，我们这里用不到那么多。

我们只用jpg原图，以及每个原图中目标的位置即可。（下边图片中画红框的两个文件夹）

但是我们发现，Annotations文件夹中的目标位置信息是存放在xml中，所以我们往下分析一个xml文件看看。

<filename> : 表示这个文件是对应于哪一个 jpg 图片的。

<size>：表示对应的 jpg 图片大小。

<object>：就是这个图片中的目标在图片中的信息。包括：目标名字，是否难识别，以及目标在整个图片中的坐标位置。（有几个 object 就是有几个目标）

提取目标初始数据

上边我们分析出每个图片中有什么目标都是存储在xml文件中的，所以我们需要将xml文件的目标与类别数据提取出来，以便我们使用。

思路：

1、使用库 xml.etree.ElementTree 读取xml格式的文件，从中提取出每一个xml文件中的所有<ojgect>标签数据（个数就是目标的数量）。
2、将<object>标签提取出 类别、xmin、ymin、xmax、ymax ，并且将其归一化为 类别、x、y、w、h。

归一化就是根据从目标中提取出的 xmin、ymin、xmax、ymax 得到目标的宽高，分别除以整个图片的宽高。

x：目标的中心位置x坐标。 y：目标的中心位置y坐标。 w：目标的宽度，h：目标的高度。

3、然后将归一化的数据按照上面的格式，整理为labels文件。

每一个labels文件对应于一个图片，labels文件中的每一行就是这个图片中的一个目标的 类别、x、y、w、h 数据（一个图片有几个目标，对应的labels文件就有几行）。

例子：

针对如下xml文件，可以得知：

对应的jpg图片是 2007_000042.jpg，并且图片的大小是 500*335 的三色图（这里的图片大小就是用来归一化的）。

含有两个<object>标签，所以这个图片中有两个目标，并且目标的类别、位置坐标可以根据name、xmin、ymin、xmax、ymax 得到。

<annotation>

	<folder>VOC2012</folder>

	<filename>2007_000042.jpg</filename>

	<source>

		<database>The VOC2007 Database</database>

		<annotation>PASCAL VOC2007</annotation>

		<image>flickr</image>

	</source>

	<size>

		<width>500</width>

		<height>335</height>

		<depth>3</depth>

	</size>

	<segmented>1</segmented>

	<object>

		<name>train</name>

		<pose>Unspecified</pose>

		<truncated>1</truncated>

		<difficult>0</difficult>

		<bndbox>

			<xmin>263</xmin>

			<ymin>32</ymin>

			<xmax>500</xmax>

			<ymax>295</ymax>

		</bndbox>

	</object>

	<object>

		<name>train</name>

		<pose>Unspecified</pose>

		<truncated>1</truncated>

		<difficult>0</difficult>

		<bndbox>

			<xmin>1</xmin>

			<ymin>36</ymin>

			<xmax>235</xmax>

			<ymax>299</ymax>

		</bndbox>

	</object>

</annotation>