yolov5 项目目录结构

636 阅读1分钟

项目目录结构

项目地址: 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  # 辅助程序代码

看到这,,,好人顺便帮忙点赞关注哈。感谢!