【目标检测】 YOLOv5算法实现(一)：算法框架概述

  本系列文章记录本人硕士阶段YOLO系列目标检测算法自学及其代码实现的过程。其中算法具体实现借鉴于ultralytics YOLO源码Github，删减了源码中部分内容，满足个人科研需求。
  本系列文章主要以YOLOv5为例完成算法的实现，后续修改、增加相关模块即可实现其他版本的YOLO算法。

文章地址：
YOLOv5算法实现(一)：算法框架概述
YOLOv5算法实现(二)：模型搭建
YOLOv5算法实现(三)：数据集构建
YOLOv5算法实现(四)：损失计算(编辑中…)
YOLOv5算法实现(五)：预测结果后处理(编辑中…)
YOLOv5算法实现(六)：评价指标及实现(编辑中…)
YOLOv5算法实现(七)：模型训练(编辑中…)
YOLOv5算法实现(八)：模型验证(编辑中…)
YOLOv5算法实现(九)：模型预测(编辑中…)

引言

  本篇文章从文件架构和算法运行逻辑两个角度对实现的YOLO算法进行概述。其中运行逻辑分别从训练、验证、测试三个角度对算法进行拆分。

文件架构

图1 算法文件架构

运行逻辑

  算法的运行逻辑如下图所示，其中虚线表示实现该功能的函数所在位置。

图2 算法运行逻辑