21. YOLOv5 代码梳理

144 阅读1分钟

学了这么久的代码了,我尝试汇总一下近段时间的思路,如有问题,后续会继续更正!保持学习!

train的整体训练架构:

图片.png

训练中模型加载过程:

图片.png

训练中数据集加载过程:

图片.png

验证中模型加载过程:

图片.png

验证中前向传播过程:

图片.png

训练中 loss 计算过程:

训练过程损失函数计算.png

验证/测试 mAP 计算过程:

验证测试流程.png