项目目录结构
项目地址: github.com/ultralytics…
整体目录
├── CONTRIBUTING.md
├── data # 配置文件夹
├── detect.py # 检测脚本
├── export.py # 模型转换脚本
├── hubconf.py # PyTorch Hub相关代码
├── LICENSE # 版权文件
├── models # yolov5模型文件
├── README.md
├── requirements.txt
├── setup.cfg
├── train.py # 模型训练脚本
├── tutorial.ipynb # Jupyter Notebook 演示代码
├── utils # yolov5的python工具文件包
└── val.py # 验证脚本
data文件
├── Argoverse.yaml
├── coco128.yaml
├── coco.yaml # COCO数据集配置文件
├── GlobalWheat2020.yaml
├── hyps # 超参数配置文件
│ ├── hyp.Objects365.yaml
│ ├── hyp.scratch-high.yaml
│ ├── hyp.scratch-low.yaml
│ ├── hyp.scratch-med.yaml
│ └── hyp.VOC.yaml
├── images
│ ├── bus.jpg
│ └── zidane.jpg
├── Objects365.yaml
├── scripts # 文件下载脚本
│ ├── download_weights.sh
│ ├── get_coco128.sh
│ └── get_coco.sh
├── SKU-110K.yaml
├── VisDrone.yaml
├── VOC.yaml
└── xView.yaml
models 文件
├── common.py # 模型组件定义代码
├── experimental.py # 实验性质的代码
├── hub # 模型拓展配置文件
│ ├── yolov5s6.yaml
│ ├── yolov5s-ghost.yaml
│ ├── yolov5s-transformer.yaml
│ └── yolov5x6.yaml
├── __init__.py
├── tf.py
├── yolo.py # Detect及Model模型构建代码
├── yolov5l.yaml # yolov5l网路配置文件
├── yolov5m.yaml
├── yolov5n.yaml
├── yolov5s.yaml
└── yolov5x.yaml
utils文件
├── activations.py # 激活函数定义代码
├── augmentations.py
├── autoanchor.py # autoanchor.py 自动锚框
├── autobatch.py
├── aws
├── benchmarks.py
├── callbacks.py
├── dataloaders.py # Dataset及Dataloader定义代码
├── docker
├── downloads.py
├── flask_rest_api
├── general.py # 项目通用函数代码
├── google_app_engine
├── __init__.py
├── loggers # 日志工具
├── loss.py # loss.py 损失函数
├── metrics.py # metrics.py性能指标
├── plots.py # Plots 画图
└── torch_utils.py # 辅助程序代码
看到这,,,好人顺便帮忙点赞关注哈。感谢!