前言
这个项目是github的开源项目,YOLOV5:https://github.com/ultralytics/yolov5,目前已经有29.4k个Star 和 10.5k 个Folk了,非常的火。下面我将会逐个的文件注释这个项目中的所有代码。
现在是2022-08-03,目标是在一个月内写完,也就是最迟2022-9-03日更完整个项目的所有的代码。
注释版全部项目文件已上传至GitHub: yolov5-6.1-annotations.
会在对源码注释上进行进一步的改进:
- 网络剪枝
- 知识蒸馏
- Grad-CAM热力图可视化
- 更换骨干网络
- 更换Neck(BiFPN,ASFF)
导航
YOLOv5-6.1
- data
- models
- common.py
- experimental.py
- tf.py
- yolo.py
- yolov5s.yaml
- utils
- activations.py
- augmentations.py
- autoanchor.py
- autobatch.py
- benchmarks.py
- callbacks.py
- datasets.py
- general.py
- loss.py
- metrics.py
- plots.py
- torch_utils.py
- detect.py
- train.py
- val.py