yolov5模型

借鉴:知乎
yolov5官方主要有4个版本,分别是Yolov5s、Yolov5m、Yolov5l和Yolov5x四个模型。

主要包括四个部分:输入端、Backbone、Neck和Prediction
(1)输入端:Mosaic数据增强、自适应锚框计算、自适应图片缩放
(2)Backbone:Focus结构、CSP结构
(3)Neck:FPN+PAN结构
(4)Prediction:GIOU_loss

整体流程如下:

1.输入图像经过骨干网络提取特征。
2.特征图通过FPN进行多尺度融合。
3.融合后的特征图通过Neck网络进行全局和局部信息整合。
4.Head网络输出目标的位置和类别。

你可能感兴趣的:(YOLO)