YOLOv5 目录
.github
.github相关，不重要
datasets
存放数据集,刚下载下来的源码是不包含这个文件夹的
classify
图像分类模块，包含以下4个文件
- predict.py —— 图像分类预测脚本
- train.py ——图像分类训练脚本
- val.py —— 图像分类验证脚本
- tutorial.ipynb —— 图像分类教程 (jupyter notebook 打开)
data
- hyps ——存放yaml格式的超参数配置文件
- images ——存放测试图片
- scripts ——存放数据集和权重下载shell脚本
runs
运行代码后生成的文件夹，存放每次实验的结构
model
- hub —— 存放yolov5各版本目标检测网络模型配置文件
- segment —— 存放yolov5各版本语义分割网络模型配置文件
- common.py——网络模型组件代码
- experimental.py——实验测试代码
- tf.py—— tensorflow版的yolov5代码
- yolo.py ——网络模型代码
- yolov5l.yaml —— 目标检测网络模型吧配置文件
- yolov5m.yaml —— 目标检测网络模型吧配置文件
- yolov5s.yaml —— 目标检测网络模型吧配置文件
- yolov5n.yaml —— 目标检测网络模型吧配置文件
- yolov5x.yaml —— 目标检测网络模型吧配置文件
segment
- predict.py ——语义分割预测脚本
- train.py—— 语义分割训练脚本
- val.py——语义分割验证脚本
- turorial.ipynb ——语义分割教程（jupyter notebook 打开）
utils
- loggers ——日志打印
- segment ——语义分割相关代码
- activations.py—— 激活函数
- autoanchor.py—— 自动生成锚框
- autobatch.py—— 自动生成批量大小
- callbacks.py —— 回调函数
- augmentations——数据增强
- metrics.py——性能计算
- plots.py ——画图
- loss.py ——损失函数
- torch_utils.py ——辅助函数
- downloads.py —— 谷歌云盘内容下载
- general.py ——相关实用函数实现
benchmarks.py —— 标准测试
train.py ——目标检测训练脚本
val.py——目标检测验证脚本
detect.py——目标检测预测脚本
export.py——模型导出
tutorial.ipynb——目标检测上手教程
hubconf.py——pytorch hub相关
requirements.txt——项目依赖的库以及版本要求
yolov5s.pt
模型预训练权重，运行代码的时候会自动从网上下载
参考文献
https://blog.csdn.net/HUASHUDEYANJING/article/details/126086708
