(10)YOLOV5算法三之各结构文件的功能作用及介绍

摘要

  1. 本文转自:YOLOV5各结构文件的功能作用
  2. yolov5 - github 地址:yolov5-master —> https://github.com/ultralytics/yolov5
  3. 关于yolov5安装、测试与训练可参考:Windows10下yolov5安装测试并训练自己的数据集
  4. 关于yolov5数据集预处理可参考:数据集转换及自动划分训练集与测试集

主题

如何下载代码,就不过多交代,网上都有,这里我们直奔主题,在对下载后的压缩包进行解压后,整个项目的文件结构入下图所示:

1  .github文件夹

.github文件夹打开后有两个文件夹如下图所示:

这两个文件夹我也不知道有什么作用,因为在用的过程中没有涉及到这两个文件,所以也没有太关注。

2  .idea文件夹

.idea文件夹打开后的目录如下图所示:

这里面的文件入门时也不是需要太关注,搞熟V5后可以研究看一下,目前我也不是很清楚。

3  data文件夹

data文件夹打开后如下图所示:

这个文件夹中,我们需要关注两个文件:coco.yaml和coco128.yaml文件,这两个文件是作者提供的数据集文件,我们在训练自己的数据集的时候,需要借用他的文件修改相应的参数,训练自己的数据集可参考(建议先看完本博客再点下面的链接):https://blog.csdn.net/oJiWuXuan/article/details/107558286  或者  https://blog.csdn.net/Clay_Zhang/article/details/107079762

以上是我们下载时的data文件夹,在我们自己训练时,需要在data文件夹中新建几个文件,我项目中的data文件夹下的目录如下图所示:

4  inference文件夹

inference文件夹只有一个images文件夹,打开如下:

images文件夹下只有两张图(应该是作者用于测试模型用的):

inference文件夹的功能主要是用于存放我们测试的图像数据,和输出的图像数据,我自己的项目中的inference文件夹目录如下图所示:

imgges文件夹用于存放需要测试图像数据,output文件夹则为对应的检测输出文件。

5   models文件夹

models文件夹打开如下:

models文件夹中主要存放了V5各模型的权重文件,在models文件夹中,还有两个.py文件是非常需要关注的,如果你需要将训练好的模型部署到c++上,就需要用到这两个文件。

6  utils文件夹

utils文件夹打开如下,主要包含了几个.py文件,这几个.py文件主要是一些函数文件。

7  weights文件夹

weights文件夹下主要用于存放初始网络权重文件,打开如下:

我们在训练网络时,需要加载作者提供的初始权重文件,这样训练出来的网络效果会更好一些。

以上主要介绍了源码目录下的文件夹,下面再介绍一下文件夹之外的文件,如下标所示:

在介绍完文件夹后,其他的几个文件,作为小白而言,主要先关注三个文件,首先是train.py和test.py文件,两个文件是用于训练网络的,这是我们入门的第一步,先调试train.py文件中的参数对网络进行训练,然后用detect.py文件,加载训练好的权重,设置需要检测的图像数据运行即可完成测试。test.py文件是在训练过程中会调其中的函数,用于在训练过程中的验证。


总结

以上是对整个V5文件目录的介绍,作为一位小白,也希望刚入门的小白看完后能有所帮助,在看完本博客后,至于如何训练自己的训练集,可参考博文:https://blog.csdn.net/oJiWuXuan/article/details/107558286  或者  https://blog.csdn.net/Clay_Zhang/article/details/107079762     亲测有用。

猜你喜欢

转载自blog.csdn.net/yohnyang/article/details/126916280